facebookresearch / ppuda

Code for Parameter Prediction for Unseen Deep Architectures (NeurIPS 2021)

MIT License

485 stars 60 forks source link

Updates #1

Closed bknyaz closed 2 years ago

bknyaz commented 2 years ago

Summary of updates

README

updated the pip installation instruction with a public repo link (now can be installed without git clone)
added a link to more checkpoints of GHNs and trained networks
added a Colab notebook and the link to the video walking through the notebook (links are added in README)
added a gif animation of the Colab notebook
added the link to the paper pdf

Code

fixed an issue with loading some pretrained networks (in fine-tuning experiments)
fixed an issue with loading pretrained GHNs from arbitrary directories