v0.4.0

riccardomusmeci released this 10 May 21:09

· 83 commits to main since this release

bd2242f

Added

Added PGNet, U2Net models
Added PGNet and MultiBCE losses
Added SaliencyDataset and SaliencyTransform utils
Added train experiment entry point
Added utils to train experiment
Added pytorch_lightning datamodule

Fixed

Correct casting between tensors in different computations

Assets 2