seems related to this:

  
    
    
    [SOLVED] Unexpected key(s) in state_dict: batches_tracked" vision
  
  
    I am getting this error when I am loading the weights for a model I trained on a GPU.  I am trying to load the model on a CPU.  Not sure what this error means.  I did check the checkpoint[‘state_dict’] keys and it does contain this ‘num_batches_tracked’ key for all the batch normalization layers.  I am using the same model on the CPU as I trained on the GPU.  I am training a UNet model. 
checkpoint = torch.load(‘model_best.pth-zeromask-notconsider.tar’,map_location=lambda storage, loc: storage) 
…