diff --git a/.gitignore b/.gitignore index 72fcf7413a2fe552e69233366f5ce86070330bf2..3a09fdec5a2d3b8ad78f8052a8a8855bef62dae9 100644 --- a/.gitignore +++ b/.gitignore @@ -7,3 +7,4 @@ Linker/__pycache__ Configuration/__pycache__ __pycache__ TensorBoard +train.py diff --git a/train.py b/train.py deleted file mode 100644 index 50354a06864fa0665d0b63927af58c976c090dd8..0000000000000000000000000000000000000000 --- a/train.py +++ /dev/null @@ -1,17 +0,0 @@ -import torch -from Configuration import Configuration -from Linker import * -from utils import read_csv_pgbar - -torch.cuda.empty_cache() -batch_size = int(Configuration.modelTrainingConfig['batch_size']) -nb_sentences = batch_size * 2 -epochs = int(Configuration.modelTrainingConfig['epoch']) - -file_path_axiom_links = 'Datasets/gold_dataset_links.csv' -df_axiom_links = read_csv_pgbar(file_path_axiom_links, nb_sentences) - -print("Linker") -linker = Linker("models/flaubert_super_98%_V2_50e.pt") -print("\nLinker Training\n\n") -linker.train_linker(df_axiom_links, validation_rate=0.1, epochs=epochs, batch_size=batch_size, checkpoint=False, tensorboard=True)