optional relu after autoregressive for compatibility
Showing
- main_sphere_compression.py 3 additions, 2 deletionsmain_sphere_compression.py
- spherical_models/compression_models/sphere_factorized_prior.py 12 additions, 5 deletions...ical_models/compression_models/sphere_factorized_prior.py
- spherical_models/compression_models/sphere_scale_hyperprior.py 11 additions, 4 deletions...ical_models/compression_models/sphere_scale_hyperprior.py
Loading
Please register or sign in to comment