-
- Downloads
implement bipartite matching loss (#1448)
* implement bipartite matching loss * fix _penv location and upgrade torch version in cicd * fix checkpoint test case after upgrading pytorch * faster matching loss * remove forward()
Showing
- .ci-cd/Dockerfile.cpu 1 addition, 1 deletion.ci-cd/Dockerfile.cpu
- alf/utils/checkpoint_utils_test.py 4 additions, 0 deletionsalf/utils/checkpoint_utils_test.py
- alf/utils/losses.py 100 additions, 1 deletionalf/utils/losses.py
- alf/utils/losses_test.py 113 additions, 0 deletionsalf/utils/losses_test.py
- setup.py 1 addition, 1 deletionsetup.py
Loading
Please register or sign in to comment