https://github.com/facebookresearch/pythia
Revision 610c1425ce2d5a4c1e4b244a594af35972c127e7 authored by ryan-qiyu-jiang on 15 September 2021, 14:11:49 UTC, committed by Facebook GitHub Bot on 15 September 2021, 14:12:49 UTC
Summary: Add loss support for classes that return a dict of losses. All loss dictionaries are flattened in the final losses dict. Added bce_kl, mse_mae loss classes as examples and for unit testing. This new loss class returns a dict, containing the bce and kl div tensors as values with string keys. Pull Request resolved: https://github.com/facebookresearch/mmf/pull/1086 Reviewed By: apsdehal Differential Revision: D30872944 Pulled By: Ryan-Qiyu-Jiang fbshipit-source-id: 4116ff92bcb24439a20d4720c3813d3c3db6ced0
1 parent ce870b1
Tip revision: 610c1425ce2d5a4c1e4b244a594af35972c127e7 authored by ryan-qiyu-jiang on 15 September 2021, 14:11:49 UTC
[feat] Add loss support for dict return type (#1086)
[feat] Add loss support for dict return type (#1086)
Tip revision: 610c142
| File | Mode | Size |
|---|---|---|
| .circleci | ||
| .github | ||
| docs | ||
| mmf | ||
| mmf_cli | ||
| projects | ||
| tests | ||
| tools | ||
| website | ||
| .editorconfig | -rw-r--r-- | 191 bytes |
| .flake8 | -rw-r--r-- | 187 bytes |
| .gitignore | -rw-r--r-- | 267 bytes |
| .pre-commit-config.yaml | -rw-r--r-- | 1.1 KB |
| LICENSE | -rw-r--r-- | 1.5 KB |
| MANIFEST.in | -rw-r--r-- | 130 bytes |
| NOTICES | -rw-r--r-- | 6.0 KB |
| README.md | -rw-r--r-- | 2.2 KB |
| pyproject.toml | -rw-r--r-- | 1017 bytes |
| requirements.txt | -rw-r--r-- | 482 bytes |
| setup.py | -rw-r--r-- | 5.1 KB |
Computing file changes ...
