Loading Code/train.py +1 −1 Original line number Original line Diff line number Diff line Loading @@ -141,7 +141,7 @@ def train(model, name, imdb, seed,mixup,lambda_value, mixepoch, tmix, mixlayer, optimizer.zero_grad() optimizer.zero_grad() model.zero_grad() model.zero_grad() torch.save(model, "./saved_models/bert_baseline.pt") torch.save(model, "saved_models/bert_baseline.pth") #evaluate trained model #evaluate trained model evaluation_test = evaluation.evaluate_model(model, name, test_dataset, learning_rate, test_batch_size, imdb) evaluation_test = evaluation.evaluate_model(model, name, test_dataset, learning_rate, test_batch_size, imdb) Loading Loading
Code/train.py +1 −1 Original line number Original line Diff line number Diff line Loading @@ -141,7 +141,7 @@ def train(model, name, imdb, seed,mixup,lambda_value, mixepoch, tmix, mixlayer, optimizer.zero_grad() optimizer.zero_grad() model.zero_grad() model.zero_grad() torch.save(model, "./saved_models/bert_baseline.pt") torch.save(model, "saved_models/bert_baseline.pth") #evaluate trained model #evaluate trained model evaluation_test = evaluation.evaluate_model(model, name, test_dataset, learning_rate, test_batch_size, imdb) evaluation_test = evaluation.evaluate_model(model, name, test_dataset, learning_rate, test_batch_size, imdb) Loading
