def train_loop(self, epochs):
    print(self.model.params)
    for i in range(int(epochs)):
        testloss = self.train()
    results = [testloss, self.model]
    return results
def train_loop(self, epochs):
    print(self.model.params)
    for i in range(int(epochs)):
        testloss = self.train()
    results = [testloss, self.model]
    return results