From 99955583ca843ed720a0f2b79dd1100c83d05efd Mon Sep 17 00:00:00 2001
From: Caroline DE POURTALES <caroline.de-pourtales@irit.fr>
Date: Mon, 18 Jul 2022 11:42:20 +0200
Subject: [PATCH] change config
---
Configuration/config.ini | 6 +++---
SuperTagger | 2 +-
train.py | 3 +--
3 files changed, 5 insertions(+), 6 deletions(-)
diff --git a/Configuration/config.ini b/Configuration/config.ini
index 17d779b..19528cd 100644
--- a/Configuration/config.ini
+++ b/Configuration/config.ini
@@ -4,7 +4,7 @@ transformers = 4.16.2
[DATASET_PARAMS]
symbols_vocab_size = 26
atom_vocab_size = 18
-max_len_sentence = 83
+max_len_sentence = 290
max_atoms_in_sentence = 900
max_atoms_in_one_type = 360
@@ -24,7 +24,7 @@ sinkhorn_iters = 5
[MODEL_TRAINING]
batch_size = 32
-pretrain_linker_epochs = 1
-epoch = 1
+pretrain_linker_epochs = 10
+epoch = 20
seed_val = 42
learning_rate = 2e-3
\ No newline at end of file
diff --git a/SuperTagger b/SuperTagger
index 4963a6b..ce978e6 160000
--- a/SuperTagger
+++ b/SuperTagger
@@ -1 +1 @@
-Subproject commit 4963a6b33a361caf3dd8c458fead6b8352b034cd
+Subproject commit ce978e6d389c6e126d162f86a4b627cda167313c
diff --git a/train.py b/train.py
index fb38519..e9b395a 100644
--- a/train.py
+++ b/train.py
@@ -6,7 +6,7 @@ from utils import read_csv_pgbar
from Configuration import Configuration
torch.cuda.empty_cache()
-nb_sentences = 4*32
+nb_sentences = 10000000
file_path_axiom_links = 'Datasets/goldANDsilver_dataset_links.csv'
model_tagger = "models/flaubert_super_98_V2_50e.pt"
@@ -28,6 +28,5 @@ print("#" * 20)
neural_proof_net = NeuralProofNet(model_tagger)
neural_proof_net.train_neuralproofnet(df_axiom_links, validation_rate=0.1, epochs=epochs, batch_size=batch_size,
checkpoint=True, tensorboard=True)
-neural_proof_net.linker.predict_without_categories("le chat est noir")
print("#" * 20)
print("#" * 20)
--
GitLab