├── .circleci └── config.yml ├── .github └── workflows │ └── ci-cd.yml ├── .gitignore ├── LICENSE ├── README.md ├── figures ├── 1.png ├── 1e3.png ├── 1e4.png ├── 2.png ├── 3e3.png ├── 3e4.png ├── 534.png ├── 5e3.png ├── 5e4.png ├── 7b.png ├── 7e4.png ├── clip_algo2.png ├── clip_percent.png ├── grad_norm_after.png ├── grad_norm_after2.png ├── grad_norm_after3.png ├── grad_norm_before.png ├── grad_norm_before2.png ├── grad_norm_before3.png ├── hist_early.png ├── hist_mid.png ├── loss_comp.png ├── low_high.png ├── lr_1e3_after.png ├── lr_1e3_before.png ├── lr_1e4_after.png ├── lr_1e4_before.png ├── lr_3e3_after.png ├── lr_3e3_before.png ├── lr_3e4_after.png ├── lr_3e4_before.png ├── lr_5e3_after.png ├── lr_5e3_before.png ├── lr_5e4_after.png ├── lr_5e4_before.png ├── lr_7e4_after.png ├── lr_7e4_before.png ├── mean.png ├── nd1.png ├── nd2.png ├── nd_1.png ├── std.png ├── through.png ├── zclip.png ├── zclip2.png ├── zclip3.png ├── zclip_poster.jpeg ├── zloss_comp.png └── zt_zstar_2_5.png ├── pyproject.toml ├── run_tests.sh ├── tests ├── conftest.py ├── test_package_config.py ├── test_zclip.py ├── test_zclip_integration.py └── test_zclip_lightning_callback.py └── zclip ├── __init__.py ├── zclip.py └── zclip_lightning_callback.py /.circleci/config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/.circleci/config.yml -------------------------------------------------------------------------------- /.github/workflows/ci-cd.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/.github/workflows/ci-cd.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/README.md -------------------------------------------------------------------------------- /figures/1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/1.png -------------------------------------------------------------------------------- /figures/1e3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/1e3.png -------------------------------------------------------------------------------- /figures/1e4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/1e4.png -------------------------------------------------------------------------------- /figures/2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/2.png -------------------------------------------------------------------------------- /figures/3e3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/3e3.png -------------------------------------------------------------------------------- /figures/3e4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/3e4.png -------------------------------------------------------------------------------- /figures/534.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/534.png -------------------------------------------------------------------------------- /figures/5e3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/5e3.png -------------------------------------------------------------------------------- /figures/5e4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/5e4.png -------------------------------------------------------------------------------- /figures/7b.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/7b.png -------------------------------------------------------------------------------- /figures/7e4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/7e4.png -------------------------------------------------------------------------------- /figures/clip_algo2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/clip_algo2.png -------------------------------------------------------------------------------- /figures/clip_percent.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/clip_percent.png -------------------------------------------------------------------------------- /figures/grad_norm_after.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/grad_norm_after.png -------------------------------------------------------------------------------- /figures/grad_norm_after2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/grad_norm_after2.png -------------------------------------------------------------------------------- /figures/grad_norm_after3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/grad_norm_after3.png -------------------------------------------------------------------------------- /figures/grad_norm_before.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/grad_norm_before.png -------------------------------------------------------------------------------- /figures/grad_norm_before2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/grad_norm_before2.png -------------------------------------------------------------------------------- /figures/grad_norm_before3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/grad_norm_before3.png -------------------------------------------------------------------------------- /figures/hist_early.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/hist_early.png -------------------------------------------------------------------------------- /figures/hist_mid.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/hist_mid.png -------------------------------------------------------------------------------- /figures/loss_comp.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/loss_comp.png -------------------------------------------------------------------------------- /figures/low_high.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/low_high.png -------------------------------------------------------------------------------- /figures/lr_1e3_after.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/lr_1e3_after.png -------------------------------------------------------------------------------- /figures/lr_1e3_before.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/lr_1e3_before.png -------------------------------------------------------------------------------- /figures/lr_1e4_after.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/lr_1e4_after.png -------------------------------------------------------------------------------- /figures/lr_1e4_before.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/lr_1e4_before.png -------------------------------------------------------------------------------- /figures/lr_3e3_after.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/lr_3e3_after.png -------------------------------------------------------------------------------- /figures/lr_3e3_before.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/lr_3e3_before.png -------------------------------------------------------------------------------- /figures/lr_3e4_after.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/lr_3e4_after.png -------------------------------------------------------------------------------- /figures/lr_3e4_before.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/lr_3e4_before.png -------------------------------------------------------------------------------- /figures/lr_5e3_after.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/lr_5e3_after.png -------------------------------------------------------------------------------- /figures/lr_5e3_before.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/lr_5e3_before.png -------------------------------------------------------------------------------- /figures/lr_5e4_after.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/lr_5e4_after.png -------------------------------------------------------------------------------- /figures/lr_5e4_before.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/lr_5e4_before.png -------------------------------------------------------------------------------- /figures/lr_7e4_after.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/lr_7e4_after.png -------------------------------------------------------------------------------- /figures/lr_7e4_before.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/lr_7e4_before.png -------------------------------------------------------------------------------- /figures/mean.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/mean.png -------------------------------------------------------------------------------- /figures/nd1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/nd1.png -------------------------------------------------------------------------------- /figures/nd2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/nd2.png -------------------------------------------------------------------------------- /figures/nd_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/nd_1.png -------------------------------------------------------------------------------- /figures/std.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/std.png -------------------------------------------------------------------------------- /figures/through.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/through.png -------------------------------------------------------------------------------- /figures/zclip.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/zclip.png -------------------------------------------------------------------------------- /figures/zclip2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/zclip2.png -------------------------------------------------------------------------------- /figures/zclip3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/zclip3.png -------------------------------------------------------------------------------- /figures/zclip_poster.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/zclip_poster.jpeg -------------------------------------------------------------------------------- /figures/zloss_comp.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/zloss_comp.png -------------------------------------------------------------------------------- /figures/zt_zstar_2_5.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/figures/zt_zstar_2_5.png -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/pyproject.toml -------------------------------------------------------------------------------- /run_tests.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/run_tests.sh -------------------------------------------------------------------------------- /tests/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/tests/conftest.py -------------------------------------------------------------------------------- /tests/test_package_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/tests/test_package_config.py -------------------------------------------------------------------------------- /tests/test_zclip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/tests/test_zclip.py -------------------------------------------------------------------------------- /tests/test_zclip_integration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/tests/test_zclip_integration.py -------------------------------------------------------------------------------- /tests/test_zclip_lightning_callback.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/tests/test_zclip_lightning_callback.py -------------------------------------------------------------------------------- /zclip/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/zclip/__init__.py -------------------------------------------------------------------------------- /zclip/zclip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/zclip/zclip.py -------------------------------------------------------------------------------- /zclip/zclip_lightning_callback.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bluorion-com/ZClip/HEAD/zclip/zclip_lightning_callback.py --------------------------------------------------------------------------------