Some questions about the code
zhukaii opened this issue · comments
Kai Zhu commented
Error when the code runs to ‘turn off’ function, should model.module be modified to the model? The pytorch model does not seem to have the '.module' attribute, only the '.modules'.
LinglanZhao commented
Line 103 in fdce946
Did you comment this line? When single GPU is available, using
model=nn.DataParallel(model,[0]
) may fix this problem.Temiloluwa Adeoti commented
0 and 1 are their GPU devices. If you have just one device, [0] will work as you described.