⚗️ | Experimenting with very small model.

2025-02-10 12:44:42 +02:00
parent 0790a0d3da
commit fb7b624c87
3 changed files with 9 additions and 10 deletions
--- a/training.py
+++ b/training.py
@ -85,7 +85,7 @@ dataset = AudioDataset(dataset_dir)

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

-train_data_loader = DataLoader(dataset, batch_size=1, shuffle=True)
+train_data_loader = DataLoader(dataset, batch_size=16, shuffle=True)

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