:albemic: | Fat architecture. Hopefully better results.

2025-04-06 00:05:43 +03:00
parent f928d8c2cf
commit 9394bc6c5a
5 changed files with 70 additions and 44 deletions
--- a/training.py
+++ b/training.py
@ -101,7 +101,7 @@ dataset = AudioDataset(dataset_dir, device)

 # ========= SINGLE =========

-train_data_loader = DataLoader(dataset, batch_size=256, shuffle=True)
+train_data_loader = DataLoader(dataset, batch_size=8, shuffle=True)

 # Initialize models and move them to device
 generator = SISUGenerator()