Solved by Narayana_Swamy in post #2 
              
              
                Looks like it was a version problem - I trained the model in 0.4.1 and tried to load the dict in 0.4.0.  Once I upgraded the pytorch version, everything worked fine.