Switch loss to bce
Showing
- dataloaders/BSDS.py 64 additions, 28 deletionsdataloaders/BSDS.py
- dataloaders/__init__.py 0 additions, 0 deletionsdataloaders/__init__.py
- models/net_builder.py 24 additions, 25 deletionsmodels/net_builder.py
- models/v1net.py 19 additions, 1 deletionmodels/v1net.py
- trainers/__init__.py 0 additions, 0 deletionstrainers/__init__.py
- trainers/base_trainer.py 38 additions, 47 deletionstrainers/base_trainer.py
- trainers/train_bsds_vanilla.py 167 additions, 56 deletionstrainers/train_bsds_vanilla.py
- utils/tf_utils.py 41 additions, 6 deletionsutils/tf_utils.py
Loading
Please register or sign in to comment