graldij / transformer-fusion

Official repository of the "Transformer Fusion with Optimal Transport" paper, published as a conference paper at ICLR 2024.
20 stars 5 forks source link

Configurations for experiments in paper. #3

Open davidleejy opened 4 months ago

davidleejy commented 4 months ago

Hi,

By any chance, would the authors be all right with sharing the configurations (and code, if any) pertaining to other datasets in this paper like CIFAR100, TINY IMAGENET, IMAGETNET-1K?

At the moment, it would appear that the one configuration file fuse_hf_vit_cifar10.yaml in this repository only applies to cifar10.

Regardless, thank you for sharing making this code base available, and congratulations on getting this work accepted at ICLR 2024.