Przeglądaj źródła

Fixed ignored neb_epochs parameter.

Kristian Schultz 3 lat temu
rodzic
commit
69f0c7a78d
1 zmienionych plików z 1 dodań i 1 usunięć
  1. 1 1
      library/generators/ConvGeN.py

+ 1 - 1
library/generators/ConvGeN.py

@@ -42,7 +42,7 @@ class ConvGeN(GanBaseClass):
         self.nebInitial = neb
         self.nebInitial = neb
         self.genInitial = gen
         self.genInitial = gen
         self.gen = gen if gen is not None else self.neb
         self.gen = gen if gen is not None else self.neb
-        self.neb_epochs = 10
+        self.neb_epochs = neb_epochs
         self.loss_history = None
         self.loss_history = None
         self.debug = debug
         self.debug = debug
         self.minSetSize = 0
         self.minSetSize = 0