rootzzp

I use this code to inference: model = models.ZharkovDilatedNet() model.load_state_dict(torch.load(ckpt_path)['model_state_dict']) heatmap = model(test_dataset)[:, 0] heatmap = torch.sigmoid(heatmap) But the download checkpoints is mismatch with the model. How to fix it?...