correct some mistakes
Merge pull request #6 from KruchDmitriy/torch

correct some mistakes