Some questions about the code

[zhukaii]

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]

CaNet/train.py

Line 103 in fdce946

model=nn.DataParallel(model,[0,1])

Did you comment this line? When single GPU is available, using model=nn.DataParallel(model,[0]) may fix this problem.

[Temiloluwa]

0 and 1 are their GPU devices. If you have just one device, [0] will work as you described.