Skip to content

Commit 8c2cd97

Browse files
committed
Merge branch 'master' into interactive_NMT
# Conflicts: # model_zoo.py
2 parents d1b8c97 + e615382 commit 8c2cd97

File tree

2 files changed

+21
-367
lines changed

2 files changed

+21
-367
lines changed

main.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,7 @@ def train_model(params):
6464

6565
# Build model
6666
if params['RELOAD'] == 0: # build new model
67-
nmt_model = TranslationModel(params, type=params['MODEL_TYPE'], verbose=params['VERBOSE'],
67+
nmt_model = TranslationModel(params, model_type=params['MODEL_TYPE'], verbose=params['VERBOSE'],
6868
model_name=params['MODEL_NAME'], vocabularies=dataset.vocabulary,
6969
store_path=params['STORE_PATH'])
7070
dict2pkl(params, params['STORE_PATH'] + '/config')
@@ -86,7 +86,7 @@ def train_model(params):
8686

8787
else: # resume from previously trained model
8888
nmt_model = TranslationModel(params,
89-
type=params['MODEL_TYPE'],
89+
model_type=params['MODEL_TYPE'],
9090
verbose=params['VERBOSE'],
9191
model_name=params['MODEL_NAME'],
9292
vocabularies=dataset.vocabulary,

0 commit comments

Comments
 (0)