Compressed model file from Channel Pruning about pocketflow HOT 10 CLOSED

@dhingratul We will, but it may take some time.

from pocketflow.

Sorry, I do not fully understand your question. What do you mean by "optimized output from tf.Saver.save() from as the .pb file"?

from pocketflow.

I am talking about the output from "Train the Compressed Model" in the tutorial, https://pocketflow.github.io/tutorial/. I am assuming as the next step is to export it as a TFLITE file, the output from the above step is a .pb file, I am looking for that file. I see you benchmarked your optimizations for Mobile Device, i am looking to do the same for the GPU, for which i need the .pb file before you generate the tflite file.

from pocketflow.

tools/conversion/export_pb_tflite_models.py generates both *.pb and *.tflite model files, located in the same directory. Can you find them in the directory specified by --model_dir?

from pocketflow.

The imagenet download is super slow, the last time I did that it took almost 3-4 days to complete. I just want to skip that and get the .pb file directly from that tutorial if possible.

from pocketflow.

Sorry, ChannelPrunedLearner requires the imagenet data set to complete its training process. Maybe you can try using smaller data sets, like CIFAR-10? Some modification will be required:

1. nets/resnet_at_ilsvrc12_run.py -> nets/resnet_at_cifar10_run.py
2. In ImageClassifierFloatResNet.java, change the image size and label file.

from pocketflow.

Yes, i understand, i was hoping that you have that you make the optimized model(after pruning) available so that i can do some benchmarks to make sure the results in terms of gains is reproducible on my machine.

from pocketflow.

Okay, we will release pre-trained compressed models for some selected pruning ratio in the next few weeks.

from pocketflow.

Enhancement required: release a few pre-trained compressed models for benchmark test.

from pocketflow.

@jiaxiang-wu Were you able to publish the pre-trained compressed models ?

from pocketflow.