Open ajinkya933 opened 3 years ago
This problem can be solved by change the following lines in the function defect_detect
, or directly use the latest version of code.
for name, module in model.named_modules():
if name == 'predictor.0':
module.register_backward_hook(func_r)
the reason is the previous name fc1
is no longer used, new model use predictor
as the new name. The hook function cannot work if it cannot find the specific layer.
@ajinkya933 Hello, would you like to ask if the polyvore-images.tar.gz data file can provide Baidu Cloud link? It won't open on github.
Duplicate of #8
@ajinkya933 Hello, would you like to ask if the polyvore-images.tar.gz data file can provide Baidu Cloud link? It won't open on github.
I am able to train model from polyvore dataset, and after training when I run diagnosis.ipynb I get this error:
Why is this error ?