search

SamsungLabs / zero-cost-nas

Zero-Cost Proxies for Lightweight NAS
Apache License 2.0
142 stars 20 forks source link

ZC proxies for 3D data and 3D CNN architectures #24

Open Junaid199f opened 12 months ago

Junaid199f commented 12 months ago

Hi! Is it possible to use these proxies for 3D CNN architectures or i have to modify these proxies to make it compatible with 3D CNN models? Thanks

vaenyr commented 12 months ago

Hi, it depends on a proxy - some of them hook specifically on Conv2d etc., in which case they would need to be extended to support 3D equivalents. Some might work out-of-the-box if they are written generically.

Particular metrics are implemented in the files under: https://github.com/SamsungLabs/zero-cost-nas/tree/main/foresight/pruners/measures