diff --git a/train.py b/train.py
index 6511a3a..eda06b6 100644
--- a/train.py
+++ b/train.py
@@ -93,8 +93,6 @@ def get_training_batch():
             writer.add_scalar('eval/lpips', eval_metrics['lpips'], epoch)
 
             to_save = models
-            to_save['opt'] = opt
-            to_save['epoch'] = epoch
 
             if eval_metrics['psnr'] > best_psnr:
                 print('best psnr model, psnr=', eval_metrics['psnr'])