Update model save path

Code/train.py

@@ -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)