Du Wei

Because the model the author trained is in torch=0.3.1, which is too low, some expressions have changed in the higher version.You can train a new model yourself and then test...

> > I solved by adding a 'strict = FALSE' on the load_state_dict function. > > line 65 of trainer.py becomes: `self.model.load_state_dict(torch.load(snapshot_file), strict=False)` > > Well, now the code is...