修改模型配置信息

Lix1993 · Aug 10, 2021 · a7fc0fd · a7fc0fd
1 parent 7739ef3
commit a7fc0fd
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/config.py b/config.py
@@ -33,7 +33,6 @@ def __init__(self):
  self.device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
  self.epochs = 10
  self.model_save_dir = os.path.join(self.project_dir, 'cache')
- self.train_info_per_batch = 30
  self.model_save_per_epoch = 2
  if not os.path.exists(self.model_save_dir):
  os.makedirs(self.model_save_dir)