Solved by Lahiru_Welagedara in post #2 
              
              
                Solution Found - 
The issue was with the torch version 1.7 in https://mathinf.eu/pytorch/arm64/. 
The model worked fine for torch version 1.6 available in https://mathinf.eu/pytorch/arm64/.