PyTorch-Encoding icon indicating copy to clipboard operation
PyTorch-Encoding copied to clipboard

Published 20 hours ago • verified publisher icon zhanghang1989

Modifying the loss and underlying models

Open nilabjodey opened this issue 5 years ago • 1 comments

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?

nilabjodey avatar Dec 04 '19 10:12 nilabjodey

Please read the docs on PyTorch for customizing models.

StacyYang avatar Dec 04 '19 16:12 StacyYang