New

KeyForce · KeyForce · commit de49c27f7c0e · 2019-09-20T20:38:40.000+08:00
diff --git a/main/train.py b/main/train.py
@@ -92,6 +92,9 @@ def fast_hist(a, b, n):
 
 
 def main():
+    # 加载数据
+    
+    
     # 使用drop_last让Batch能够整除
     train_loader = torch.utils.data.DataLoader(train_data, batch_size=16, drop_last=True)
     test_loader = torch.utils.data.DataLoader(test_data, batch_size=16, drop_last=True)
