├── .gitignore ├── README.md ├── assets ├── rot_trick.png └── rot_trick_logo.jpeg ├── environment.yaml ├── exp_weights_and_configs ├── table_3 │ ├── 32_32_4 │ │ └── config.yaml │ ├── 64_64_3 │ │ └── config.yaml │ └── rot_vqgan.py └── table_4 │ ├── config.yaml │ └── rot_vitvqgan.py └── src ├── __init__.py ├── data ├── dataloader.py └── imagenet_dataset.py ├── eval_utils └── eval_loop.py ├── local_vector_quantize_pytorch └── vector_quantize_pytorch.py ├── models ├── model_utils.py └── vq_vae.py ├── modules ├── blocks.py ├── decoder.py └── encoder.py ├── reconstruction_fid.py ├── reconstruction_is.py ├── scripts.sh ├── train.py └── train_utils ├── schedulers.py ├── train_loop.py ├── trainer.py ├── utils.py └── wandb_utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cfifty/rotation_trick/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cfifty/rotation_trick/HEAD/README.md -------------------------------------------------------------------------------- /assets/rot_trick.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cfifty/rotation_trick/HEAD/assets/rot_trick.png -------------------------------------------------------------------------------- /assets/rot_trick_logo.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cfifty/rotation_trick/HEAD/assets/rot_trick_logo.jpeg -------------------------------------------------------------------------------- /environment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cfifty/rotation_trick/HEAD/environment.yaml -------------------------------------------------------------------------------- /exp_weights_and_configs/table_3/32_32_4/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cfifty/rotation_trick/HEAD/exp_weights_and_configs/table_3/32_32_4/config.yaml -------------------------------------------------------------------------------- /exp_weights_and_configs/table_3/64_64_3/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cfifty/rotation_trick/HEAD/exp_weights_and_configs/table_3/64_64_3/config.yaml -------------------------------------------------------------------------------- /exp_weights_and_configs/table_3/rot_vqgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cfifty/rotation_trick/HEAD/exp_weights_and_configs/table_3/rot_vqgan.py -------------------------------------------------------------------------------- /exp_weights_and_configs/table_4/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cfifty/rotation_trick/HEAD/exp_weights_and_configs/table_4/config.yaml -------------------------------------------------------------------------------- /exp_weights_and_configs/table_4/rot_vitvqgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cfifty/rotation_trick/HEAD/exp_weights_and_configs/table_4/rot_vitvqgan.py -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/data/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cfifty/rotation_trick/HEAD/src/data/dataloader.py -------------------------------------------------------------------------------- /src/data/imagenet_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cfifty/rotation_trick/HEAD/src/data/imagenet_dataset.py -------------------------------------------------------------------------------- /src/eval_utils/eval_loop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cfifty/rotation_trick/HEAD/src/eval_utils/eval_loop.py -------------------------------------------------------------------------------- /src/local_vector_quantize_pytorch/vector_quantize_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cfifty/rotation_trick/HEAD/src/local_vector_quantize_pytorch/vector_quantize_pytorch.py -------------------------------------------------------------------------------- /src/models/model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cfifty/rotation_trick/HEAD/src/models/model_utils.py -------------------------------------------------------------------------------- /src/models/vq_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cfifty/rotation_trick/HEAD/src/models/vq_vae.py -------------------------------------------------------------------------------- /src/modules/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cfifty/rotation_trick/HEAD/src/modules/blocks.py -------------------------------------------------------------------------------- /src/modules/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cfifty/rotation_trick/HEAD/src/modules/decoder.py -------------------------------------------------------------------------------- /src/modules/encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cfifty/rotation_trick/HEAD/src/modules/encoder.py -------------------------------------------------------------------------------- /src/reconstruction_fid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cfifty/rotation_trick/HEAD/src/reconstruction_fid.py -------------------------------------------------------------------------------- /src/reconstruction_is.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cfifty/rotation_trick/HEAD/src/reconstruction_is.py -------------------------------------------------------------------------------- /src/scripts.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cfifty/rotation_trick/HEAD/src/scripts.sh -------------------------------------------------------------------------------- /src/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cfifty/rotation_trick/HEAD/src/train.py -------------------------------------------------------------------------------- /src/train_utils/schedulers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cfifty/rotation_trick/HEAD/src/train_utils/schedulers.py -------------------------------------------------------------------------------- /src/train_utils/train_loop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cfifty/rotation_trick/HEAD/src/train_utils/train_loop.py -------------------------------------------------------------------------------- /src/train_utils/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cfifty/rotation_trick/HEAD/src/train_utils/trainer.py -------------------------------------------------------------------------------- /src/train_utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cfifty/rotation_trick/HEAD/src/train_utils/utils.py -------------------------------------------------------------------------------- /src/train_utils/wandb_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cfifty/rotation_trick/HEAD/src/train_utils/wandb_utils.py --------------------------------------------------------------------------------