diff --git a/Train.py b/Train.py index cac81e28eda1701970c597e5d54bcac7d8e98ec0..cd218748eca9db025bb1862455a2ecc94e9e6bd3 100644 --- a/Train.py +++ b/Train.py @@ -62,7 +62,7 @@ def extractExamples(debug, ts, strat, config, dicts, network, dynamic) : example = torch.cat([torch.LongTensor([goldIndex]), features]) examples.append(example) - moved = applyTransition(ts, strat, config, candidate) + moved = applyTransition(ts, strat, config, candidate, None) EOS.apply(config, strat)