Some questions about the code
zhukaii opened this issue · comments
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'.
Line 103 in fdce946
Did you comment this line? When single GPU is available, using
model=nn.DataParallel(model,[0]
) may fix this problem.0 and 1 are their GPU devices. If you have just one device, [0] will work as you described.