├── LICENSE ├── README.md ├── dataset ├── README.md ├── __init__.py ├── generate_pretrained_model_park_data.py ├── generate_simple_cnn_park_data.py ├── pretrained_model_park │ ├── __init__.py │ ├── hyperparameters.csv │ ├── interface.py │ └── models │ │ ├── __init__.py │ │ ├── basic_cnn.py │ │ ├── basic_cnn_1d.py │ │ ├── deepsets.py │ │ ├── ffcv_model.py │ │ ├── resnet.py │ │ ├── small_resnet.py │ │ ├── transformer.py │ │ └── utils.py └── simple_cnn_park │ ├── __init__.py │ ├── hyperparameters.csv │ └── interface.py ├── experiments ├── __init__.py ├── checkpoints │ └── dyhpo_transfer ├── fms.py ├── gmn │ ├── encoders.py │ ├── feature_extractor_gmn.py │ ├── graph_construct │ │ ├── __init__.py │ │ ├── constants.py │ │ ├── layers.py │ │ ├── model_arch_graph.ipynb │ │ ├── model_arch_graph.py │ │ ├── net_makers.py │ │ └── utils.py │ ├── graph_models.py │ ├── graph_pooling.py │ └── grit.py ├── logme.py ├── nfn │ ├── __init__.py │ ├── cnn_example.py │ ├── common │ │ ├── __init__.py │ │ └── data.py │ ├── helpers.py │ └── layers │ │ ├── __init__.py │ │ ├── encoding.py │ │ ├── layer_utils.py │ │ ├── layers.py │ │ ├── misc_layers.py │ │ └── regularize.py ├── results │ ├── make_plots.py │ ├── make_smooth_plots.py │ ├── plot_ablations.py │ ├── plot_best_finetuning.py │ └── plot_transferrability.py ├── scripts │ ├── dyhpo.sh │ ├── fms-flat.sh │ ├── fms-gmn.sh │ ├── fms-nfn.sh │ └── random-search.sh ├── surrogates.py └── utils.py ├── figures └── fms_overview.png ├── requirements.txt └── setup.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/README.md -------------------------------------------------------------------------------- /dataset/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/dataset/README.md -------------------------------------------------------------------------------- /dataset/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataset/generate_pretrained_model_park_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/dataset/generate_pretrained_model_park_data.py -------------------------------------------------------------------------------- /dataset/generate_simple_cnn_park_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/dataset/generate_simple_cnn_park_data.py -------------------------------------------------------------------------------- /dataset/pretrained_model_park/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataset/pretrained_model_park/hyperparameters.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/dataset/pretrained_model_park/hyperparameters.csv -------------------------------------------------------------------------------- /dataset/pretrained_model_park/interface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/dataset/pretrained_model_park/interface.py -------------------------------------------------------------------------------- /dataset/pretrained_model_park/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataset/pretrained_model_park/models/basic_cnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/dataset/pretrained_model_park/models/basic_cnn.py -------------------------------------------------------------------------------- /dataset/pretrained_model_park/models/basic_cnn_1d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/dataset/pretrained_model_park/models/basic_cnn_1d.py -------------------------------------------------------------------------------- /dataset/pretrained_model_park/models/deepsets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/dataset/pretrained_model_park/models/deepsets.py -------------------------------------------------------------------------------- /dataset/pretrained_model_park/models/ffcv_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/dataset/pretrained_model_park/models/ffcv_model.py -------------------------------------------------------------------------------- /dataset/pretrained_model_park/models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/dataset/pretrained_model_park/models/resnet.py -------------------------------------------------------------------------------- /dataset/pretrained_model_park/models/small_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/dataset/pretrained_model_park/models/small_resnet.py -------------------------------------------------------------------------------- /dataset/pretrained_model_park/models/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/dataset/pretrained_model_park/models/transformer.py -------------------------------------------------------------------------------- /dataset/pretrained_model_park/models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/dataset/pretrained_model_park/models/utils.py -------------------------------------------------------------------------------- /dataset/simple_cnn_park/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataset/simple_cnn_park/hyperparameters.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/dataset/simple_cnn_park/hyperparameters.csv -------------------------------------------------------------------------------- /dataset/simple_cnn_park/interface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/dataset/simple_cnn_park/interface.py -------------------------------------------------------------------------------- /experiments/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /experiments/checkpoints/dyhpo_transfer: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/experiments/checkpoints/dyhpo_transfer -------------------------------------------------------------------------------- /experiments/fms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/experiments/fms.py -------------------------------------------------------------------------------- /experiments/gmn/encoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/experiments/gmn/encoders.py -------------------------------------------------------------------------------- /experiments/gmn/feature_extractor_gmn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/experiments/gmn/feature_extractor_gmn.py -------------------------------------------------------------------------------- /experiments/gmn/graph_construct/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /experiments/gmn/graph_construct/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/experiments/gmn/graph_construct/constants.py -------------------------------------------------------------------------------- /experiments/gmn/graph_construct/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/experiments/gmn/graph_construct/layers.py -------------------------------------------------------------------------------- /experiments/gmn/graph_construct/model_arch_graph.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/experiments/gmn/graph_construct/model_arch_graph.ipynb -------------------------------------------------------------------------------- /experiments/gmn/graph_construct/model_arch_graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/experiments/gmn/graph_construct/model_arch_graph.py -------------------------------------------------------------------------------- /experiments/gmn/graph_construct/net_makers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/experiments/gmn/graph_construct/net_makers.py -------------------------------------------------------------------------------- /experiments/gmn/graph_construct/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/experiments/gmn/graph_construct/utils.py -------------------------------------------------------------------------------- /experiments/gmn/graph_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/experiments/gmn/graph_models.py -------------------------------------------------------------------------------- /experiments/gmn/graph_pooling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/experiments/gmn/graph_pooling.py -------------------------------------------------------------------------------- /experiments/gmn/grit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/experiments/gmn/grit.py -------------------------------------------------------------------------------- /experiments/logme.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/experiments/logme.py -------------------------------------------------------------------------------- /experiments/nfn/__init__.py: -------------------------------------------------------------------------------- 1 | from nfn import layers, common -------------------------------------------------------------------------------- /experiments/nfn/cnn_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/experiments/nfn/cnn_example.py -------------------------------------------------------------------------------- /experiments/nfn/common/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/experiments/nfn/common/__init__.py -------------------------------------------------------------------------------- /experiments/nfn/common/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/experiments/nfn/common/data.py -------------------------------------------------------------------------------- /experiments/nfn/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/experiments/nfn/helpers.py -------------------------------------------------------------------------------- /experiments/nfn/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/experiments/nfn/layers/__init__.py -------------------------------------------------------------------------------- /experiments/nfn/layers/encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/experiments/nfn/layers/encoding.py -------------------------------------------------------------------------------- /experiments/nfn/layers/layer_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/experiments/nfn/layers/layer_utils.py -------------------------------------------------------------------------------- /experiments/nfn/layers/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/experiments/nfn/layers/layers.py -------------------------------------------------------------------------------- /experiments/nfn/layers/misc_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/experiments/nfn/layers/misc_layers.py -------------------------------------------------------------------------------- /experiments/nfn/layers/regularize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/experiments/nfn/layers/regularize.py -------------------------------------------------------------------------------- /experiments/results/make_plots.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/experiments/results/make_plots.py -------------------------------------------------------------------------------- /experiments/results/make_smooth_plots.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/experiments/results/make_smooth_plots.py -------------------------------------------------------------------------------- /experiments/results/plot_ablations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/experiments/results/plot_ablations.py -------------------------------------------------------------------------------- /experiments/results/plot_best_finetuning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/experiments/results/plot_best_finetuning.py -------------------------------------------------------------------------------- /experiments/results/plot_transferrability.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/experiments/results/plot_transferrability.py -------------------------------------------------------------------------------- /experiments/scripts/dyhpo.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/experiments/scripts/dyhpo.sh -------------------------------------------------------------------------------- /experiments/scripts/fms-flat.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/experiments/scripts/fms-flat.sh -------------------------------------------------------------------------------- /experiments/scripts/fms-gmn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/experiments/scripts/fms-gmn.sh -------------------------------------------------------------------------------- /experiments/scripts/fms-nfn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/experiments/scripts/fms-nfn.sh -------------------------------------------------------------------------------- /experiments/scripts/random-search.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/experiments/scripts/random-search.sh -------------------------------------------------------------------------------- /experiments/surrogates.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/experiments/surrogates.py -------------------------------------------------------------------------------- /experiments/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/experiments/utils.py -------------------------------------------------------------------------------- /figures/fms_overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/figures/fms_overview.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Forecasting-Model-Search/HEAD/setup.py --------------------------------------------------------------------------------