Skip to content
Snippets Groups Projects
Commit 5ce419a1 authored by kulcsar's avatar kulcsar
Browse files

path changes in main

parent 8ef10463
No related branches found
No related tags found
No related merge requests found
......@@ -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)
......
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
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment