mirror of
https://github.com/facebookresearch/fairseq.git
synced 2024-09-19 05:09:20 +03:00
Make error message for trying to train after make_generation_fast work correctly
Summary: https://github.com/pytorch/fairseq/blob/master/fairseq/trainer.py#L164 calls `train()` without any argument Reviewed By: myleott Differential Revision: D13599203 fbshipit-source-id: 3a096a6dd35a7a3f8309fbda3b54a36f606475e3
This commit is contained in:
parent
4b1f4788d8
commit
315fa5cbd9
@ -117,7 +117,7 @@ class BaseFairseqModel(nn.Module):
|
|||||||
|
|
||||||
self.apply(apply_make_generation_fast_)
|
self.apply(apply_make_generation_fast_)
|
||||||
|
|
||||||
def train(mode):
|
def train(mode=True):
|
||||||
if mode:
|
if mode:
|
||||||
raise RuntimeError('cannot train after make_generation_fast')
|
raise RuntimeError('cannot train after make_generation_fast')
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user