change ignore

.gitignore

@@ -7,3 +7,4 @@ Linker/__pycache__
 Configuration/__pycache__
 __pycache__
 TensorBoard
+train.py

train.py

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