├── .gitignore ├── LICENSE ├── README.md ├── configs ├── aistats21 │ ├── figure7_classification │ │ ├── f7_c10_wrn_baseline.yml │ │ ├── f7_c10_wrn_sttpR_r008.yml │ │ ├── f7_c10_wrn_sttpR_r016.yml │ │ ├── f7_c10_wrn_sttpR_r032.yml │ │ ├── f7_c10_wrn_sttpR_r064.yml │ │ ├── f7_c10_wrn_sttpR_r128.yml │ │ ├── f7_c10_wrn_sttpR_r256.yml │ │ ├── f7_c10_wrn_svdpR_r004.yml │ │ ├── f7_c10_wrn_svdpR_r008.yml │ │ ├── f7_c10_wrn_svdpR_r016.yml │ │ ├── f7_c10_wrn_svdpR_r032.yml │ │ ├── f7_c10_wrn_svdpR_r064.yml │ │ └── f7_c10_wrn_svdpR_r128.yml │ ├── table1_gan_reduced_discriminator │ │ ├── t1_c10u_sngan_sn.yml │ │ ├── t1_c10u_sngan_sr.yml │ │ ├── t1_c10u_sngan_sttpC.yml │ │ ├── t1_c10u_sngan_svdpC.yml │ │ ├── t1_s10u_sngan_sn.yml │ │ ├── t1_s10u_sngan_sr.yml │ │ ├── t1_s10u_sngan_sttpC.yml │ │ └── t1_s10u_sngan_svdpC.yml │ ├── table2_gan_full_discriminator │ │ ├── t2_c10u_sngan_sn.yml │ │ ├── t2_c10u_sngan_sr.yml │ │ ├── t2_c10u_sngan_sttpC.yml │ │ ├── t2_c10u_sngan_sttpL.yml │ │ ├── t2_c10u_sngan_sttpR.yml │ │ ├── t2_c10u_sngan_svdpC.yml │ │ ├── t2_c10u_sngan_svdpL.yml │ │ ├── t2_c10u_sngan_svdpR.yml │ │ ├── t2_s10u_sngan_sn.yml │ │ ├── t2_s10u_sngan_sr.yml │ │ ├── t2_s10u_sngan_sttpC.yml │ │ ├── t2_s10u_sngan_sttpL.yml │ │ ├── t2_s10u_sngan_sttpR.yml │ │ ├── t2_s10u_sngan_svdpC.yml │ │ ├── t2_s10u_sngan_svdpL.yml │ │ └── t2_s10u_sngan_svdpR.yml │ └── table3_figure6_gan_rank │ │ ├── t3_c10u_sngan_sttpR_r016.yml │ │ ├── t3_c10u_sngan_sttpR_r024.yml │ │ ├── t3_c10u_sngan_sttpR_r032.yml │ │ ├── t3_c10u_sngan_sttpR_r048.yml │ │ ├── t3_c10u_sngan_sttpR_r064.yml │ │ ├── t3_c10u_sngan_sttpR_r096.yml │ │ ├── t3_c10u_sngan_sttpR_r128.yml │ │ ├── t3_c10u_sngan_svdpR_r004.yml │ │ ├── t3_c10u_sngan_svdpR_r008.yml │ │ ├── t3_c10u_sngan_svdpR_r016.yml │ │ ├── t3_c10u_sngan_svdpR_r024.yml │ │ ├── t3_c10u_sngan_svdpR_r032.yml │ │ ├── t3_c10u_sngan_svdpR_r048.yml │ │ └── t3_c10u_sngan_svdpR_r064.yml └── env_lsf.yml ├── doc ├── AISTATS21_Obukhov_STTP_paper.pdf ├── AISTATS21_Obukhov_STTP_poster.pdf ├── config.md ├── poster.png ├── requirements_reproducibility.txt ├── teaser.gif └── teaser.mp4 ├── requirements.txt └── src ├── datasets └── torchvision.py ├── modules ├── losses.py ├── model_resnet_cifar10.py ├── model_wideresnet_cifar.py ├── ops_spectral_norm_pytorch.py └── sngan.py ├── test └── test_tt_modules.py ├── train.py ├── train_gan.py ├── train_imgcls.py └── utils ├── config.py ├── helpers.py ├── resolvers.py ├── spectral_compensation.py ├── spectral_penalties.py ├── spectral_tensors_factory.py ├── stiefel_parameterization.py ├── tensor_contraction.py ├── tt_utils.py └── visualizations.py /.gitignore: -------------------------------------------------------------------------------- 1 | .DS_Store 2 | .idea 3 | __pycache__ 4 | src/scripts 5 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/README.md -------------------------------------------------------------------------------- /configs/aistats21/figure7_classification/f7_c10_wrn_baseline.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/figure7_classification/f7_c10_wrn_baseline.yml -------------------------------------------------------------------------------- /configs/aistats21/figure7_classification/f7_c10_wrn_sttpR_r008.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/figure7_classification/f7_c10_wrn_sttpR_r008.yml -------------------------------------------------------------------------------- /configs/aistats21/figure7_classification/f7_c10_wrn_sttpR_r016.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/figure7_classification/f7_c10_wrn_sttpR_r016.yml -------------------------------------------------------------------------------- /configs/aistats21/figure7_classification/f7_c10_wrn_sttpR_r032.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/figure7_classification/f7_c10_wrn_sttpR_r032.yml -------------------------------------------------------------------------------- /configs/aistats21/figure7_classification/f7_c10_wrn_sttpR_r064.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/figure7_classification/f7_c10_wrn_sttpR_r064.yml -------------------------------------------------------------------------------- /configs/aistats21/figure7_classification/f7_c10_wrn_sttpR_r128.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/figure7_classification/f7_c10_wrn_sttpR_r128.yml -------------------------------------------------------------------------------- /configs/aistats21/figure7_classification/f7_c10_wrn_sttpR_r256.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/figure7_classification/f7_c10_wrn_sttpR_r256.yml -------------------------------------------------------------------------------- /configs/aistats21/figure7_classification/f7_c10_wrn_svdpR_r004.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/figure7_classification/f7_c10_wrn_svdpR_r004.yml -------------------------------------------------------------------------------- /configs/aistats21/figure7_classification/f7_c10_wrn_svdpR_r008.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/figure7_classification/f7_c10_wrn_svdpR_r008.yml -------------------------------------------------------------------------------- /configs/aistats21/figure7_classification/f7_c10_wrn_svdpR_r016.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/figure7_classification/f7_c10_wrn_svdpR_r016.yml -------------------------------------------------------------------------------- /configs/aistats21/figure7_classification/f7_c10_wrn_svdpR_r032.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/figure7_classification/f7_c10_wrn_svdpR_r032.yml -------------------------------------------------------------------------------- /configs/aistats21/figure7_classification/f7_c10_wrn_svdpR_r064.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/figure7_classification/f7_c10_wrn_svdpR_r064.yml -------------------------------------------------------------------------------- /configs/aistats21/figure7_classification/f7_c10_wrn_svdpR_r128.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/figure7_classification/f7_c10_wrn_svdpR_r128.yml -------------------------------------------------------------------------------- /configs/aistats21/table1_gan_reduced_discriminator/t1_c10u_sngan_sn.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/table1_gan_reduced_discriminator/t1_c10u_sngan_sn.yml -------------------------------------------------------------------------------- /configs/aistats21/table1_gan_reduced_discriminator/t1_c10u_sngan_sr.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/table1_gan_reduced_discriminator/t1_c10u_sngan_sr.yml -------------------------------------------------------------------------------- /configs/aistats21/table1_gan_reduced_discriminator/t1_c10u_sngan_sttpC.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/table1_gan_reduced_discriminator/t1_c10u_sngan_sttpC.yml -------------------------------------------------------------------------------- /configs/aistats21/table1_gan_reduced_discriminator/t1_c10u_sngan_svdpC.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/table1_gan_reduced_discriminator/t1_c10u_sngan_svdpC.yml -------------------------------------------------------------------------------- /configs/aistats21/table1_gan_reduced_discriminator/t1_s10u_sngan_sn.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/table1_gan_reduced_discriminator/t1_s10u_sngan_sn.yml -------------------------------------------------------------------------------- /configs/aistats21/table1_gan_reduced_discriminator/t1_s10u_sngan_sr.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/table1_gan_reduced_discriminator/t1_s10u_sngan_sr.yml -------------------------------------------------------------------------------- /configs/aistats21/table1_gan_reduced_discriminator/t1_s10u_sngan_sttpC.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/table1_gan_reduced_discriminator/t1_s10u_sngan_sttpC.yml -------------------------------------------------------------------------------- /configs/aistats21/table1_gan_reduced_discriminator/t1_s10u_sngan_svdpC.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/table1_gan_reduced_discriminator/t1_s10u_sngan_svdpC.yml -------------------------------------------------------------------------------- /configs/aistats21/table2_gan_full_discriminator/t2_c10u_sngan_sn.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/table2_gan_full_discriminator/t2_c10u_sngan_sn.yml -------------------------------------------------------------------------------- /configs/aistats21/table2_gan_full_discriminator/t2_c10u_sngan_sr.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/table2_gan_full_discriminator/t2_c10u_sngan_sr.yml -------------------------------------------------------------------------------- /configs/aistats21/table2_gan_full_discriminator/t2_c10u_sngan_sttpC.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/table2_gan_full_discriminator/t2_c10u_sngan_sttpC.yml -------------------------------------------------------------------------------- /configs/aistats21/table2_gan_full_discriminator/t2_c10u_sngan_sttpL.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/table2_gan_full_discriminator/t2_c10u_sngan_sttpL.yml -------------------------------------------------------------------------------- /configs/aistats21/table2_gan_full_discriminator/t2_c10u_sngan_sttpR.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/table2_gan_full_discriminator/t2_c10u_sngan_sttpR.yml -------------------------------------------------------------------------------- /configs/aistats21/table2_gan_full_discriminator/t2_c10u_sngan_svdpC.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/table2_gan_full_discriminator/t2_c10u_sngan_svdpC.yml -------------------------------------------------------------------------------- /configs/aistats21/table2_gan_full_discriminator/t2_c10u_sngan_svdpL.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/table2_gan_full_discriminator/t2_c10u_sngan_svdpL.yml -------------------------------------------------------------------------------- /configs/aistats21/table2_gan_full_discriminator/t2_c10u_sngan_svdpR.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/table2_gan_full_discriminator/t2_c10u_sngan_svdpR.yml -------------------------------------------------------------------------------- /configs/aistats21/table2_gan_full_discriminator/t2_s10u_sngan_sn.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/table2_gan_full_discriminator/t2_s10u_sngan_sn.yml -------------------------------------------------------------------------------- /configs/aistats21/table2_gan_full_discriminator/t2_s10u_sngan_sr.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/table2_gan_full_discriminator/t2_s10u_sngan_sr.yml -------------------------------------------------------------------------------- /configs/aistats21/table2_gan_full_discriminator/t2_s10u_sngan_sttpC.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/table2_gan_full_discriminator/t2_s10u_sngan_sttpC.yml -------------------------------------------------------------------------------- /configs/aistats21/table2_gan_full_discriminator/t2_s10u_sngan_sttpL.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/table2_gan_full_discriminator/t2_s10u_sngan_sttpL.yml -------------------------------------------------------------------------------- /configs/aistats21/table2_gan_full_discriminator/t2_s10u_sngan_sttpR.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/table2_gan_full_discriminator/t2_s10u_sngan_sttpR.yml -------------------------------------------------------------------------------- /configs/aistats21/table2_gan_full_discriminator/t2_s10u_sngan_svdpC.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/table2_gan_full_discriminator/t2_s10u_sngan_svdpC.yml -------------------------------------------------------------------------------- /configs/aistats21/table2_gan_full_discriminator/t2_s10u_sngan_svdpL.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/table2_gan_full_discriminator/t2_s10u_sngan_svdpL.yml -------------------------------------------------------------------------------- /configs/aistats21/table2_gan_full_discriminator/t2_s10u_sngan_svdpR.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/table2_gan_full_discriminator/t2_s10u_sngan_svdpR.yml -------------------------------------------------------------------------------- /configs/aistats21/table3_figure6_gan_rank/t3_c10u_sngan_sttpR_r016.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/table3_figure6_gan_rank/t3_c10u_sngan_sttpR_r016.yml -------------------------------------------------------------------------------- /configs/aistats21/table3_figure6_gan_rank/t3_c10u_sngan_sttpR_r024.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/table3_figure6_gan_rank/t3_c10u_sngan_sttpR_r024.yml -------------------------------------------------------------------------------- /configs/aistats21/table3_figure6_gan_rank/t3_c10u_sngan_sttpR_r032.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/table3_figure6_gan_rank/t3_c10u_sngan_sttpR_r032.yml -------------------------------------------------------------------------------- /configs/aistats21/table3_figure6_gan_rank/t3_c10u_sngan_sttpR_r048.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/table3_figure6_gan_rank/t3_c10u_sngan_sttpR_r048.yml -------------------------------------------------------------------------------- /configs/aistats21/table3_figure6_gan_rank/t3_c10u_sngan_sttpR_r064.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/table3_figure6_gan_rank/t3_c10u_sngan_sttpR_r064.yml -------------------------------------------------------------------------------- /configs/aistats21/table3_figure6_gan_rank/t3_c10u_sngan_sttpR_r096.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/table3_figure6_gan_rank/t3_c10u_sngan_sttpR_r096.yml -------------------------------------------------------------------------------- /configs/aistats21/table3_figure6_gan_rank/t3_c10u_sngan_sttpR_r128.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/table3_figure6_gan_rank/t3_c10u_sngan_sttpR_r128.yml -------------------------------------------------------------------------------- /configs/aistats21/table3_figure6_gan_rank/t3_c10u_sngan_svdpR_r004.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/table3_figure6_gan_rank/t3_c10u_sngan_svdpR_r004.yml -------------------------------------------------------------------------------- /configs/aistats21/table3_figure6_gan_rank/t3_c10u_sngan_svdpR_r008.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/table3_figure6_gan_rank/t3_c10u_sngan_svdpR_r008.yml -------------------------------------------------------------------------------- /configs/aistats21/table3_figure6_gan_rank/t3_c10u_sngan_svdpR_r016.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/table3_figure6_gan_rank/t3_c10u_sngan_svdpR_r016.yml -------------------------------------------------------------------------------- /configs/aistats21/table3_figure6_gan_rank/t3_c10u_sngan_svdpR_r024.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/table3_figure6_gan_rank/t3_c10u_sngan_svdpR_r024.yml -------------------------------------------------------------------------------- /configs/aistats21/table3_figure6_gan_rank/t3_c10u_sngan_svdpR_r032.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/table3_figure6_gan_rank/t3_c10u_sngan_svdpR_r032.yml -------------------------------------------------------------------------------- /configs/aistats21/table3_figure6_gan_rank/t3_c10u_sngan_svdpR_r048.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/table3_figure6_gan_rank/t3_c10u_sngan_svdpR_r048.yml -------------------------------------------------------------------------------- /configs/aistats21/table3_figure6_gan_rank/t3_c10u_sngan_svdpR_r064.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/aistats21/table3_figure6_gan_rank/t3_c10u_sngan_svdpR_r064.yml -------------------------------------------------------------------------------- /configs/env_lsf.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/configs/env_lsf.yml -------------------------------------------------------------------------------- /doc/AISTATS21_Obukhov_STTP_paper.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/doc/AISTATS21_Obukhov_STTP_paper.pdf -------------------------------------------------------------------------------- /doc/AISTATS21_Obukhov_STTP_poster.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/doc/AISTATS21_Obukhov_STTP_poster.pdf -------------------------------------------------------------------------------- /doc/config.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/doc/config.md -------------------------------------------------------------------------------- /doc/poster.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/doc/poster.png -------------------------------------------------------------------------------- /doc/requirements_reproducibility.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/doc/requirements_reproducibility.txt -------------------------------------------------------------------------------- /doc/teaser.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/doc/teaser.gif -------------------------------------------------------------------------------- /doc/teaser.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/doc/teaser.mp4 -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/requirements.txt -------------------------------------------------------------------------------- /src/datasets/torchvision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/src/datasets/torchvision.py -------------------------------------------------------------------------------- /src/modules/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/src/modules/losses.py -------------------------------------------------------------------------------- /src/modules/model_resnet_cifar10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/src/modules/model_resnet_cifar10.py -------------------------------------------------------------------------------- /src/modules/model_wideresnet_cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/src/modules/model_wideresnet_cifar.py -------------------------------------------------------------------------------- /src/modules/ops_spectral_norm_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/src/modules/ops_spectral_norm_pytorch.py -------------------------------------------------------------------------------- /src/modules/sngan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/src/modules/sngan.py -------------------------------------------------------------------------------- /src/test/test_tt_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/src/test/test_tt_modules.py -------------------------------------------------------------------------------- /src/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/src/train.py -------------------------------------------------------------------------------- /src/train_gan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/src/train_gan.py -------------------------------------------------------------------------------- /src/train_imgcls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/src/train_imgcls.py -------------------------------------------------------------------------------- /src/utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/src/utils/config.py -------------------------------------------------------------------------------- /src/utils/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/src/utils/helpers.py -------------------------------------------------------------------------------- /src/utils/resolvers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/src/utils/resolvers.py -------------------------------------------------------------------------------- /src/utils/spectral_compensation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/src/utils/spectral_compensation.py -------------------------------------------------------------------------------- /src/utils/spectral_penalties.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/src/utils/spectral_penalties.py -------------------------------------------------------------------------------- /src/utils/spectral_tensors_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/src/utils/spectral_tensors_factory.py -------------------------------------------------------------------------------- /src/utils/stiefel_parameterization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/src/utils/stiefel_parameterization.py -------------------------------------------------------------------------------- /src/utils/tensor_contraction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/src/utils/tensor_contraction.py -------------------------------------------------------------------------------- /src/utils/tt_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/src/utils/tt_utils.py -------------------------------------------------------------------------------- /src/utils/visualizations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/toshas/sttp/HEAD/src/utils/visualizations.py --------------------------------------------------------------------------------