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

[vovantuanuit]

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