Possible bug: Channel Pruning + RL with MobileNetV1

Tencent / PocketFlow

An Automatic Model Compression (AutoMC) framework for developing smaller and faster AI applications.

https://pocketflow.github.io

Other

2.78k stars 490 forks source link

Possible bug: Channel Pruning + RL with MobileNetV1 #163

Open GoldenSpark opened 5 years ago

GoldenSpark commented 5 years ago

Hanging during the pruning.

extra_args: --learner channel --cp_preserve_ratio 0.5 --cp_prune_option auto --ddpg_noise_std_init=0.5 --ddpg_actor_width 300 --ddpg_critic_width 300 --cp_nb_rlouts_min 100 --cp_nb_rlouts 400
--cp_finetune=False --cp_retrain=True --enbl_dst=True --batch_size_eval 64 --data_dir_local /data/NewTFRecord/

mobilev1_img_cp_auto_0.5_noise0.5_width300_roll100-400_save2_4GPU.log

jiaxiang-wu commented 5 years ago

Possibly duplicate with issue #160 (training hang after some iterations). We are looking into this.