I was able to installed apex but when I tried with ResNet50 pretrained keras weights, it was asking to specify all the hyper-parameters so while prunning it will take refernce from pretrained base model. However, I have mentioned but it was failing and throwing AssertionError. Can you guys provide some sample script or guide that how to apply ASP on model zoo pretrained models?
I have resolved that error but now when I am trying to perform ASP again it was taking more than 20hrs but not getting prunned weights.
ASP.prune_trained_model(model, optimizer)
After above line, It is taking infinite time. Can you pls help to assist me how to apply ASP on pretrained model?
Based on the description of the ASP repository, it targets distributed training with PyTorch.
However, PyTorch distributed training might implement with CUDA IPC which is not available on Jetson.
It’s recommended to check if ASP can work on Jetson first.
Thanks.