Fix resume checkpoint mapping for torch models only (#370)

Fix resume checkpoint mapping for torch models only (#370) #163