-
- Downloads
Implement the MixtureProjectionNetwork (#1411)
* Unify ParallelNormalProjectionNetwork and NormalProjectionNetwork * Add make_parallel() back for normal projection network * Add parallel FC version of make_parallel to BetaProjectionNetwork * Implement the MixtureProjectionNetwork * Address comments
Showing
- alf/networks/projection_networks.py 74 additions, 0 deletionsalf/networks/projection_networks.py
- alf/networks/projection_networks_test.py 71 additions, 0 deletionsalf/networks/projection_networks_test.py
- alf/utils/dist_utils.py 17 additions, 0 deletionsalf/utils/dist_utils.py
- alf/utils/dist_utils_test.py 17 additions, 0 deletionsalf/utils/dist_utils_test.py
Loading
Please register or sign in to comment