boahk/media_cdarl

Official repository for "C-DARL: Contrastive diffusion adversarial representation learning for label-free blood vessel segmentation"

media_cdarl's Issues

Unexpected Key Error while resuming training

I am facing an error using your model while resuming training at any epoch, if i do strict=False in the same it says AttributeError: 'CDARL' object has no attribute 'optH'
Could you provide any resolution of it
in load_network
network.load_state_dict(torch.load(H_path), strict=(not self.opt['model']['finetune_norm']))
File "/opt/conda/envs/cdarl/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1667, in load_state_dict
raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format(
RuntimeError: Error(s) in loading state_dict for PatchSampleF:
Unexpected key(s) in state_dict: "mlp_0.0.weight", "mlp_0.0.bias", "mlp_0.2.weight", "mlp_0.2.bias", "mlp_1.0.weight", "mlp_1.0.bias", "mlp_1.2.weight", "mlp_1.2.bias", "mlp_2.0.weight", "mlp_2.0.bias", "mlp_2.2.weight", "mlp_2.2.bias".