CaNet icon indicating copy to clipboard operation
CaNet copied to clipboard

Published 20 hours ago • verified publisher icon icoz69

Some questions about the code

Open zhukaii opened this issue 5 years ago • 2 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'.

zhukaii avatar Sep 15 '19 05:09 zhukaii

https://github.com/icoz69/CaNet/blob/fdce9462d03ff93bc11f4795df94fbbf60b1c8ca/train.py#L103 Did you comment this line? When single GPU is available, using model=nn.DataParallel(model,[0]) may fix this problem.

LinglanZhao avatar Jun 11 '20 11:06 LinglanZhao

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

Temiloluwa avatar Oct 25 '20 20:10 Temiloluwa