Update surrogate.py

modules/surrogate.py

@@ -70,7 +70,7 @@ class Surrogate(torch.nn.Module):
         # take into account that 
         self.layers = torch.nn.Sequential(
             #torch.nn.BatchNorm1d(n_detector_parameters + n_true_context_inputs + n_reco_parameters + 1),
-            torch.nn.Linear(n_detector_parameters + n_true_context_inputs + n_reco_parameters + 1 , 100), #1 for time step
+            torch.nn.Linear(n_detector_parameters + n_true_context_inputs + n_reco_parameters + 1 , 100), #1 for time step, removed + 1 for including cost in the OG parameters too
             torch.nn.ELU(),
             #torch.nn.BatchNorm1d(100),
             torch.nn.Linear(100, 100),