@@ -42,7 +42,7 @@ class ConvGeN(GanBaseClass):
self.nebInitial = neb
self.genInitial = gen
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.debug = debug
self.minSetSize = 0