All models running.

2019-08-24 19:05:46 +02:00
parent 7b0b96eaa3
commit 18305a9e7e
6 changed files with 45 additions and 45 deletions
--- a/networks/modules.py
+++ b/networks/modules.py
@ -25,9 +25,9 @@ class LightningModuleOverrides:

    @data_loader
    def tng_dataloader(self):
-        num_workers = os.cpu_count() // 2
+        num_workers = 0  #  os.cpu_count() // 2
        return DataLoader(DataContainer('data', self.size, self.step),
-                          shuffle=True, batch_size=100, num_workers=num_workers)
+                          shuffle=True, batch_size=10000, num_workers=num_workers)


 class AbstractNeuralNetwork(Module):