wang

@Dav-Jay , I came cross this bug when I used the code for a customized data set. My patch for the bug is as follows: ``` def forward(self, x): x...