train

2026-08-24 09:40:31 +08:00
parent 5199e3040f
commit 2bf7db6bb9
23 changed files with 51 additions and 105 deletions
@@ -76,4 +76,5 @@ print("test_dataset", len(test_dataset))


 train_dataloader = DataLoader(dataset=train_dataset, batch_size=batch_size, shuffle=True, num_workers=4)
-test_dataloader = DataLoader(dataset=test_dataset, batch_size=batch_size, shuffle=False, num_workers=4)
+test_dataloader = DataLoader(dataset=test_dataset, batch_size=batch_size, shuffle=False, num_workers=4)
+