search

Wizaron / instance-segmentation-pytorch

Semantic Instance Segmentation with a Discriminative Loss Function in PyTorch
https://arxiv.org/abs/1708.02551
GNU General Public License v3.0
458 stars 95 forks source link

Fix parameter passing bug. #17

Open nieyan opened 4 years ago

nieyan commented 4 years ago

If training with usegpu=False , will throw assert error of size_average.