Modifying the loss and underlying models

zhanghang1989 / PyTorch-Encoding

A CV toolkit for my papers.

https://hangzhang.org/PyTorch-Encoding/

MIT License

2.04k stars 452 forks source link

Modifying the loss and underlying models #236

Open nilabjodey opened 4 years ago

nilabjodey commented 4 years ago

How can I train a new model while modifying the loss function used in the encnet as well as the deeplab / fcn.. model cross-entropy loss ? Also, how can I remove encnet from the structure and do an ablation study?

StacyYang commented 4 years ago

Please read the docs on PyTorch for customizing models.