From 70f73e24de19d249f67b661a521183c640900f69 Mon Sep 17 00:00:00 2001
From: PNRIA - Julien <julien.rabault@irit.fr>
Date: Thu, 19 May 2022 15:52:40 +0200
Subject: [PATCH] change ignore

---
 .gitignore |  1 +
 train.py   | 17 -----------------
 2 files changed, 1 insertion(+), 17 deletions(-)
 delete mode 100644 train.py

diff --git a/.gitignore b/.gitignore
index 72fcf74..3a09fde 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 50354a0..0000000
--- 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)
-- 
GitLab