Hi, I’m the OP. Thank you for your reply. This has been resolved. Check this out if it helps.

  
      github.com/caffe2/models
  
  
    



Issue: Mobilenet_v2_quantized predict_net.pb file possible error 9


	
	opened by TerryTsao
	on 2018-11-14
	
	
	


TL;DR
Can't run the net using caffe2 following the 'Load Pretrained Net' tutorial. Keep showing error saying that 325 is not CPU...




  
  
    
    
  
  


The problem with the official model is that 325 refers to an Int8Tensor, which cannot be the output of a network for some reason I’m not familiar with.
Also, in case you are using any pytorch version not containing this PR from me (https://github.com/pytorch/pytorch/pull/15047 7), you might experience a bug related to fetching Int8Tensor. It seems the fix is not included in the latest v1.0.1 yet. So if you happens to encounter some weird issue, there’s a simple fix.
HTH.