def __init__(self, model, numIntType, energy, dim):
     self.model = MultiLayerNet(model[0], model[1], model[2])
     self.model = self.model.to(dev)
     self.intLoss = IntegrationLoss(numIntType, dim)
     self.energy = energy
     self.lossArray = []
     self.dim = dim
 def __init__(self, model, numIntType, energy, dim):
     # self.data = data
     self.model = MultiLayerNet(model[0], model[1], model[2])
     self.model = self.model.to(dev)
     self.intLoss = IntegrationLoss(numIntType, dim)
     self.energy = energy
     # self.post = PostProcessing(energy, dim)
     self.dim = dim
 def __init__(self, model, numIntType, energy, dim):
     # self.data = data
     self.model = MultiLayerNet(model[0], model[1], model[2])
     self.model = self.model.to(dev)
     self.intLoss = IntegrationLoss(numIntType, dim)
     self.energy = EnergyModel(energy, dim)