binary-nets icon indicating copy to clipboard operation
binary-nets copied to clipboard

Model can't be moved to another device after inference

Open mzemlyanikin opened this issue 6 years ago • 0 comments

If we create a model on cpu and run it, than copies of convolutions' weights (weight.original) are saved on cpu. If we run model.cuda(), they stay on cpu.

mzemlyanikin avatar Feb 20 '19 20:02 mzemlyanikin