bonlime / pytorch-tools

Tool box for PyTorch
MIT License
186 stars 16 forks source link

add soft CCE loss #39

Closed bonlime closed 5 years ago

bonlime commented 5 years ago

Добавил Categorical CrossEntropy с опцией smoothing. В случае smoothing=0 этот лосс эквивалентен встроенному torch.nn.CrossEntropyLoss. Единственное важное отличие - эта версия требует на вход либо one-hot encoded labels, либо указать количество классов при инициализации, что отличается от дефолтного поведения версии из торча. Это сделано для того, чтобы можно было из коробки поддерживать мультикласс классификацию, когда у каждого объекта несколько классов.