pytorch

가중치 업데이트 연산 과정에서 어떠한 장비를 선택할지에 대한 코드이다. CUDA를 통해 GPU를 사용할 수 있다면 torch.cuda.is.available()에 True값이, 사용할 수 없다면 False 값이 저장된다. 이에 따라 device에 'cuda' 또는 'c