Update train.py

This commit is contained in:
Yifan Yang 2023-05-31 11:45:17 +08:00 committed by GitHub
parent 2036652598
commit 9d751b9af4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -516,7 +516,6 @@ def train_one_epoch(
model_avg=model_avg,
)
if (
params.batch_idx_train > 0
and params.batch_idx_train % params.save_every_n == 0