Dev-repo_cleanup models and losses docstrings

neptune-ai / open-solution-mapping-challenge

Open solution to the Mapping Challenge :earth_americas:

https://www.crowdai.org/challenges/mapping-challenge

MIT License

380 stars 96 forks source link

Dev-repo_cleanup models and losses docstrings #135

Closed taraspiotr closed 6 years ago

taraspiotr commented 6 years ago

Minor reformatting in docstrings
Refactored loss functions in models.py
Docstrings for ResNet and VGG16 U-Net models
Docstrings for loss functions in models.py

taraspiotr commented 6 years ago

@jakubczakon @apyskir I was not sure, if losses docstrings should contain exact definitions of loss functions, since they are only wrappers for lower level loss functions, where they are (or should be) documented in detail, for example in nn.CrossEntropy(). What do you think?