forgot to divide checkpoint freq by iterations to get checkpoint counts

This commit is contained in:
mrq 2023-02-19 07:05:11 +00:00
parent 485319c2bb
commit d89b7d60e0

View File

@ -465,7 +465,7 @@ def run_training(config_path, verbose=False, buffer_size=8, progress=gr.Progress
its = config['train']['niter']
checkpoint = 0
checkpoints = config['logger']['save_checkpoint_freq']
checkpoints = config['logger']['save_checkpoint_freq'] / its
buffer_size = 8
open_state = False