diff --git a/trainer.py b/trainer.py index 51a05e6..ae9f0e4 100644 --- a/trainer.py +++ b/trainer.py @@ -20,7 +20,6 @@ def __init__(self, online_network, target_network, predictor, optimizer, device, self.m = params['m'] self.batch_size = params['batch_size'] self.num_workers = params['num_workers'] - self.checkpoint_interval = params['checkpoint_interval'] _create_model_training_folder(self.writer, files_to_same=["./config/config.yaml", "main.py", 'trainer.py']) @torch.no_grad()