├── .gitignore ├── CITATION.cff ├── README.md ├── configurations ├── configs.py └── sweep_config.yml ├── dev.env ├── environment.yml ├── main_multigpu.py ├── main_singlegpu.py ├── model_dicts └── vgg_variants.pkl ├── models ├── custom_frft_layers.py ├── resnet.py ├── resnet_pretrained.py ├── vgg.py └── vgg_pretrained.py ├── pyproject.toml ├── requirements.txt ├── sweep_singlegpu.py ├── trainers ├── multigpu_trainer.py └── trainer.py ├── utils ├── data_utils.py ├── main_utils.py ├── trainer_utils.py └── utils.py └── vgg_variants.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/koc-lab/TrainableFrFT/HEAD/.gitignore -------------------------------------------------------------------------------- /CITATION.cff: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/koc-lab/TrainableFrFT/HEAD/CITATION.cff -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/koc-lab/TrainableFrFT/HEAD/README.md -------------------------------------------------------------------------------- /configurations/configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/koc-lab/TrainableFrFT/HEAD/configurations/configs.py -------------------------------------------------------------------------------- /configurations/sweep_config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/koc-lab/TrainableFrFT/HEAD/configurations/sweep_config.yml -------------------------------------------------------------------------------- /dev.env: -------------------------------------------------------------------------------- 1 | WANDB_API_KEY=XXX 2 | -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/koc-lab/TrainableFrFT/HEAD/environment.yml -------------------------------------------------------------------------------- /main_multigpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/koc-lab/TrainableFrFT/HEAD/main_multigpu.py -------------------------------------------------------------------------------- /main_singlegpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/koc-lab/TrainableFrFT/HEAD/main_singlegpu.py -------------------------------------------------------------------------------- /model_dicts/vgg_variants.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/koc-lab/TrainableFrFT/HEAD/model_dicts/vgg_variants.pkl -------------------------------------------------------------------------------- /models/custom_frft_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/koc-lab/TrainableFrFT/HEAD/models/custom_frft_layers.py -------------------------------------------------------------------------------- /models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/koc-lab/TrainableFrFT/HEAD/models/resnet.py -------------------------------------------------------------------------------- /models/resnet_pretrained.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/koc-lab/TrainableFrFT/HEAD/models/resnet_pretrained.py -------------------------------------------------------------------------------- /models/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/koc-lab/TrainableFrFT/HEAD/models/vgg.py -------------------------------------------------------------------------------- /models/vgg_pretrained.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/koc-lab/TrainableFrFT/HEAD/models/vgg_pretrained.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/koc-lab/TrainableFrFT/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/koc-lab/TrainableFrFT/HEAD/requirements.txt -------------------------------------------------------------------------------- /sweep_singlegpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/koc-lab/TrainableFrFT/HEAD/sweep_singlegpu.py -------------------------------------------------------------------------------- /trainers/multigpu_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/koc-lab/TrainableFrFT/HEAD/trainers/multigpu_trainer.py -------------------------------------------------------------------------------- /trainers/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/koc-lab/TrainableFrFT/HEAD/trainers/trainer.py -------------------------------------------------------------------------------- /utils/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/koc-lab/TrainableFrFT/HEAD/utils/data_utils.py -------------------------------------------------------------------------------- /utils/main_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/koc-lab/TrainableFrFT/HEAD/utils/main_utils.py -------------------------------------------------------------------------------- /utils/trainer_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/koc-lab/TrainableFrFT/HEAD/utils/trainer_utils.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/koc-lab/TrainableFrFT/HEAD/utils/utils.py -------------------------------------------------------------------------------- /vgg_variants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/koc-lab/TrainableFrFT/HEAD/vgg_variants.py --------------------------------------------------------------------------------