Rooba
/
genienlp
mirror of https://github.com/stanford-oval/genienlp.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

change start iteration for resume

This commit is contained in:
Bryan Marcus McCann 2018-06-24 14:34:02 +00:00
parent 913ee7368e
commit faa1ad382e
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -142,7 +142,7 @@ def train(args, model, opt, train_iters, train_iterations, field, rank=0, world_
"""main training function""" """main training function"""
logger = log(rank) logger = log(rank)
local_loss, num_examples, len_contexts, len_answers, iteration = 0, 0, 0, 0, 1 local_loss, num_examples, len_contexts, len_answers, iteration = 0, 0, 0, 0, start_iteration
train_iter_deep = deepcopy(train_iterations) train_iter_deep = deepcopy(train_iterations)
local_train_metric_dict = {} local_train_metric_dict = {}