Skip to content
forked from TalonCB/MANAS

Modularized Adaptive Neural Architecture Search

License

Notifications You must be signed in to change notification settings

rutgerswiselab/MANAS

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

27 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

MANAS

This repository includes the implementation for Learning Basic Skills and Reuse: Modularized Adaptive Neural Architecture Search (MANAS):

Hanxiong Chen, Yunqi Li, He Zhu, Yongfeng Zhang. 2022.

Refernece

For inquiries contact Hanxiong Chen (hanxiong.chen@rutgers.edu) or Yongfeng Zhang (yongfeng.zhang@rutgers.edu)

Environments

Python>=3.6
numpy>=1.18.1
pytorch>=1.4.0
pandas>=1.1.0
scipy>=1.3.0
tqdm>=4.32.1
scikit_learn>=0.23.1

Datasets

  • The processed datasets are in ./dataset/

  • Amazon Datasets: The origin dataset can be found here.

Example to run the codes

  • To guarantee the program can execute properly, please keep the directory structure as given in this repository. The parameter configuration file is located at ./src/config.py
  • Example run:
# MANAS on Cellphone dataset
> cd ./src/
> python ./main.py --dataset Cellphone --child_model_path "../model/Cellphone_child_model.pt" --controller_model_path "../model/Cellphone_controller_model.pt"

About

Modularized Adaptive Neural Architecture Search

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%