MLNet-Pytorch
MLNet-Pytorch copied to clipboard
Published
20 hours ago •
immortal3
immortal3
Implementation of A Deep Multi-Level Network for Saliency Prediction in Pytorch
Results
2
MLNet-Pytorch issues
Sort by
recently updated
recently updated
newest added
when i load your model, it gives error: RuntimeError: Error(s) in loading state_dict for MLNet: Missing key(s) in state_dict: "prior". code i am using: PATH = "./mlnet/pretrain.model" model.load_state_dict(torch.load(PATH,map_location=torch.device('cpu'))) thanks in...
Thanks for your great work! But there seems to be an incorrect implementation for the ModMSELoss. 1) The division operation `(output / output_max)` could cause `NaN` values if all entries...
