from local
This commit is contained in:
parent
c801e9e454
commit
29167ca9dc
Binary file not shown.
Binary file not shown.
@ -1038,6 +1038,7 @@ def train_one_epoch(
|
|||||||
set_batch_count(model, params.batch_idx_train)
|
set_batch_count(model, params.batch_idx_train)
|
||||||
logging.info('4')
|
logging.info('4')
|
||||||
scheduler.step_batch(params.batch_idx_train)
|
scheduler.step_batch(params.batch_idx_train)
|
||||||
|
print(optimizer)
|
||||||
logging.info('5')
|
logging.info('5')
|
||||||
scaler.step(optimizer)
|
scaler.step(optimizer)
|
||||||
logging.info('6')
|
logging.info('6')
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user