-
Notifications
You must be signed in to change notification settings - Fork 492
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
I can't run "Automatic Channel Pruning" on local mode . #52
Comments
Hi, could you please post the full log file? |
./scripts/run_seven.sh nets/resnet_at_cifar10_run.py Python script: nets/resnet_at_cifar10_run.py of GPUs: 1extra arguments: --learner channel --batch_size_eval 64 --cp_preserve_ratio 0.5 --cp_prune_option auto --resnet_size 20 ['model_http_url', 'https://api.ai.tencent.com/pocketflow']
INFO:tensorflow:RL choosed preserv ratio: [[ 0.28163278]] |
Hi, are you running the program with |
yes I have change scripts/run_seven.sh to scripts/run_local.sh ,the first lines of the log were added by myself to say which command I have used |
Can you post the full content of your |
the shell scripts is: default argumentsnb_gpus=1 parse arguments passed from the command linepy_script="$1" obtain list of idle GPUsidle_gpus= re-create the logging directoryrm -rf logs && mkdir logs execute the specified Python script with one or more GPUscp -v ${py_script} main.py The command is: |
hi @sunzhe09 @jiaxiang-wu, I am running into the exact issue when I tried the auto mode on the Imagenet dataset on a local machine with one gpu card. Run command Error Trace
Potential Root Cause # lines between 695-700 of learners/channel_pruning/learner.py
# in function __prune_rl()
with self.pruner.model.g.as_default():
self.__save_in_progress_pruned_model() In the function # lines between 480-485 of learners/channel_pruning/learner.py
def __save_in_progress_pruned_model(self):
""" save a in progress training model with a max evaluation result"""
self.max_save_path = self.saver_eval.save(self.sess_eval, FLAGS.cp_best_path)
tf.logging.info('model saved best model to ' + self.max_save_path) |
@psyyz10 Can you take a look at this issue? |
Resolved in this pull request: #65 |
this bug still exists! After pull request: #65 merged! |
@psyyz10 Can you keep an eye on this issue? |
Actually, remove code from pull request: #65, bug fixed! |
I am a outside user of tencent, so,I change the run_seven.sh to run_local.sh ,but when it save the progress model ,it said Nontype,I think there isn't any model has been generated.Does it only support seven mode?
The text was updated successfully, but these errors were encountered: