pytorch-SceneNetRGBD icon indicating copy to clipboard operation
pytorch-SceneNetRGBD copied to clipboard

Published 20 hours ago • verified publisher icon ankurhanda

Trainning model

Open littlerain2310 opened this issue 2 years ago • 0 comments

How can I train your pretrained model ?

You said your code based on https://github.com/milesial/Pytorch-UNet, but when I loaded your model and try to predict some picture it gave me error Missing key(s) in state_dict: "inc.double_conv.0.weight", "inc.double_conv.1.weight" ,etc. How you train your model ?

littlerain2310 avatar Sep 27 '22 05:09 littlerain2310