Elias Manos

Hi Roberto, Did you end up getting it to work? If so, would you mind sharing with me how you created the custom dataset and inferenced with the pre-trained model?...

Well, I managed to figure this out by modifying `optimizer_state_dict` with the following: ``` ckpt_mod['optimizer_state_dict']['state'][196]['exp_avg'] = checkpoint['optimizer_state_dict']['state'][196]['exp_avg'][np.r_[2,5,6]] ckpt_mod['optimizer_state_dict']['state'][196]['exp_avg_sq'] = checkpoint['optimizer_state_dict']['state'][196]['exp_avg_sq'][np.r_[2,5,6]] ckpt_mod['optimizer_state_dict']['state'][197]['exp_avg'] = checkpoint['optimizer_state_dict']['state'][197]['exp_avg'][np.r_[2,5,6]] ckpt_mod['optimizer_state_dict']['state'][197]['exp_avg_sq'] = checkpoint['optimizer_state_dict']['state'][197]['exp_avg_sq'][np.r_[2,5,6]] ``` Now I'm running...

I'm having the same issue. Have you figured out the cause or a solution?

I figured out the issue on my end. Try placing an underscore between the i and j numbers in the filename. Something like this: ``` # Save patches into new...

I am getting the same issue as I try to implement this into my UNet model for image segmentation. I know it has been a while but have you found...