Skip to content

Large-scale analysis of pretrained deep neural networks reveals parametric relationship between model similarity and model attributes.

Notifications You must be signed in to change notification settings

VandyCatLab/modelzoo

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Model Zoo

This repository requires an annoying combination of very specific dependencies to ensure that Tensorflow and Pytorch (and their respect model hubs) will work together.

Installation

pip install -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cu118

The requirements.txt probably works but if it doesn't, manually installing the important packages should work, provided the exact right versions are installed.

Note that we use torch==2.3.1+cu118 and tensorflow==2.14.1 because both of these use CUDA 11.8, which should be supported by the nvidia-driver-545. CUDA 11.8 should still probably be manually installed to avoid issues of dependencies crashing together (even if installing Tensorflow with the CUDA extras tensorflow[and-cuda]==2.14.1). Importantly, Tensorflow Hub must be pinned for installation (e.g., tensorflow-hub==0.15.0) as later version will not work as it depends on the latest version of Keras, which is not compatible with our specific version of tensorflow.

About

Large-scale analysis of pretrained deep neural networks reveals parametric relationship between model similarity and model attributes.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 97.5%
  • Python 2.5%