Update train.py

pre-committed
pull/2386/head
Zhao Yuting 3 years ago committed by GitHub
parent 5821bbbb44
commit aeaf297f63
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -137,7 +137,7 @@ if __name__ == '__main__':
if epoch % config[
'save_freq'] == 0 and batch_idx + 1 == steps_per_epoch and local_rank == 0:
num_corrects = 0
num_samples = 0

Loading…
Cancel
Save