Deep-SAD-PyTorch icon indicating copy to clipboard operation
Deep-SAD-PyTorch copied to clipboard

Correctly load state from pretraining

Open davnn opened this issue 4 years ago • 1 comments

    # Filter out decoder network keys
    ae_net_dict = {k: v for k, v in ae_net_dict.items() if k in net_dict}

Did not match any keys. The encoders seem to live in .encoder of the corresponding AE and we can directly import the keys from there.

davnn avatar Jan 31 '21 09:01 davnn

addresses https://github.com/lukasruff/Deep-SAD-PyTorch/issues/9

davnn avatar Jan 31 '21 10:01 davnn