microsoft / Cream

This is a collection of our NAS and Vision Transformer work.
MIT License
1.61k stars 220 forks source link

How to print the subnet architectures get from supernet AutoFormer (tiny,base,small)? #231

Open vovantuanuit opened 3 months ago

vovantuanuit commented 3 months ago

Hi, first thanks for the nice work! It's really interesting,

Currently, I did like to print out the architectures of AutoFormer subnet, however, when I print the subnet architectures, the subnet is the get from super-net like this: model_subnet = unwrap_model(model_VIT_supernet) model_subnet.set_sample_config(config=net_arch) model_subnet.to(device)

the output architectures of subnet when i just print(model_subnet) is always the architectures of supernet.

Is there any other way to print out the architectures of subnet? Thanks you

Best regards, Tuan