search

unibas-gravis / Occlusion-Robust-MoFA

Other
207 stars 30 forks source link

Problem in first train step #23

Closed NewtOliver closed 5 months ago

NewtOliver commented 9 months ago

How should I solve this problem? My cuda version is 11.3, python version is 3.8
image

yug125lk commented 9 months ago

change this line in networks.py : feature = self.net(image, is_shallow) to this: feature = self.net(image)