Closed XiaoqiangZhou closed 4 years ago
Hi I've come across the same issue. Were you able to solve it?
Hi, these are calls for internal functions in the nn.Module class (the models are extending it), it sets the mode as training or validation (issues when using BatchNorm and Dropout).
@hedjm Thanks for your reply. I have got it. And I will close this issue.
Hello, I'm recently reading you codes, which are written in a good manner.
But I'm confused with the following code in edge_connect.py line 99 and line 100:
self.edge_model.train()
self.inpaint_model.train()
There is notrain()
attribute inclass EdgeModel
andclass InpaintingModel
. Could you please explain what do these two codes mean? Thanks!