Commit ef32f86f authored by friebolin's avatar friebolin
Browse files

Update model save path

parent 5be6cc4d

Code/train.py

+1 −1
Original line number Diff line number Diff line
@@ -141,7 +141,7 @@ def train(model, name, imdb, seed,mixup,lambda_value, mixepoch, tmix, mixlayer, 
					optimizer.zero_grad()

					model.zero_grad()

	

	torch.save(model, "./saved_models/bert_baseline.pt")

	torch.save(model, "saved_models/bert_baseline.pth")



	#evaluate trained model

	evaluation_test = evaluation.evaluate_model(model, name,  test_dataset, learning_rate, test_batch_size, imdb)