PocketFlow
PocketFlow copied to clipboard
Possible bug: Channel Pruning + RL with MobileNetV1
trafficstars
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
Possibly duplicate with issue #160 (training hang after some iterations). We are looking into this.