from local

This commit is contained in:
dohe0342 2022-12-26 15:22:58 +09:00
parent c801e9e454
commit 29167ca9dc
3 changed files with 1 additions and 0 deletions

Binary file not shown.

View File

@ -1038,6 +1038,7 @@ def train_one_epoch(
set_batch_count(model, params.batch_idx_train)
logging.info('4')
scheduler.step_batch(params.batch_idx_train)
print(optimizer)
logging.info('5')
scaler.step(optimizer)
logging.info('6')