As mentioned in #5,
The output of "layer_4" is NaN for Line 71 in modules/geo_predictors/omnidata/modules/midas/dpt_depth.py
_layer_1, layer_2, layer_3, layer_4 = forwardvit(self.pretrained, x)
The pre-trained checkpoint seems to load successfully for the normal prediction model.
How to fix the issue? thanks!
As mentioned in #5, The output of "layer_4" is NaN for Line 71 in modules/geo_predictors/omnidata/modules/midas/dpt_depth.py _layer_1, layer_2, layer_3, layer_4 = forwardvit(self.pretrained, x)
The pre-trained checkpoint seems to load successfully for the normal prediction model. How to fix the issue? thanks!