model

Softcatala · Oct 1, 2024 · 1a909c3 · 1a909c3
1 parent 7603ce4
commit 1a909c3
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 21 deletions.
diff --git a/languages/master-language/model.py b/languages/master-language/model.py
diff --git a/languages/master-language/train.sh b/languages/master-language/train.sh
@@ -1,2 +1,2 @@
 #--num_gpus 4 for -> multiple gpus
-onmt-main --mixed_precision --model model.py --config data.yml --auto_config train --with_eval
+onmt-main --mixed_precision --model_type TransformerBigRelative --config data.yml --auto_config train --with_eval