path changes in main

Code/train.py

@@ -144,8 +144,8 @@ def train(model, name, imdb, seed,mixup,lambda_value, mixepoch, tmix, mixlayer,
 	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)
-	evaluation_train = evaluation.evaluate_model(model, name, train_dataset, learning_rate, test_batch_size, imdb)
+	evaluation_test = evaluation.evaluate_model(model, name,  test_dataset, test_batch_size, imdb)
+	evaluation_train = evaluation.evaluate_model(model, name, train_dataset, test_batch_size, imdb)
  
 	print("TEST: ", evaluation_test)
 	print("TRAIN: ", evaluation_train)

main.py

 import argparse
-import preprocess
-import train
-import evaluation
-import models
+import Code.preprocess
+import Code.train
+import Code.evaluation
+import Code.models
 import json
 import copy
 from transformers import BertTokenizer, RobertaTokenizer, BertModel, RobertaModel, RobertaPreTrainedModel, RobertaConfig,  BertConfig, BertPreTrainedModel, PreTrainedModel, AutoConfig, AutoModel, AutoTokenizer