Разрешение PyTorch RuntimeError: ожидаемый скалярный тип double, но найдено число с плавающей запятой