Network cleanup

network/base_network.py

@@ -12,7 +12,7 @@ import numpy as np
 
 # Base class for all kinds of network
 class BasicNet:
-    def __init__(self, optimizer_fn, gpu, LSTM=False):
+    def __init__(self, gpu, LSTM=False):
         self.gpu = gpu and torch.cuda.is_available()
         self.LSTM = LSTM
         if self.gpu: