diff --git a/sevenn/scripts/processing_by_batch.py b/sevenn/scripts/processing_by_batch.py index 92534d7d..3fb7501f 100644 --- a/sevenn/scripts/processing_by_batch.py +++ b/sevenn/scripts/processing_by_batch.py @@ -158,7 +158,7 @@ def processing_by_batch( train_loader.sampler.refresh_sequence() # make starting index to zero trainer.write_checkpoint( - f'{prefix}/checkpoint_initial.pth', + f'{prefix}/checkpoint_0.pth', config=config, epoch=0, data_progress=data_progress,