Run

python train_ppo_fbs_static_dynamic.py --budget 0.5 --dpratio 0.5

Requirements

pytorch >= 1.4

Name		Name	Last commit message	Last commit date
Latest commit History 1 Commit
utils		utils
BudgetPruning_static_dynamic.py		BudgetPruning_static_dynamic.py
PPOAgentGaussian.py		PPOAgentGaussian.py
PPOAgentGaussian_static.py		PPOAgentGaussian_static.py
README.md		README.md
hook_cifarnet_static_dynamic_agg.py		hook_cifarnet_static_dynamic_agg.py
train_ppo_fbs_static_dynamic.py		train_ppo_fbs_static_dynamic.py