목록2024/07/29 (1)
플로라도의 data workout
PyTorch cuda에서 CPU추론으로 변경할때 확인해야 할 사항
1.모델1-1. 가중치 로드 model.load(weight_path, map_location=device) # 코드 확인 1-2. 모델 객체 이동 확인 모델객체.to(device) # 코드 확인 1-3. 모델 평가모드 설정with torch.no_grad() 루프와 model.eval()모드 확인 2. tensor이동'data(tensor).to(device)' 코드 확인 3. nn.DataParallel 삭제멀티-GPU 환경에서 학습한 스크립트 일부를 이용할시, 추론단계에서는 'nn.DataParallel' 관련 코드 삭제 model.to('cuda')model = nn.DataParallel(model)model.to('cuda') 과 의 model의 state_dict의 key값은 상이하..
기초 노트/PyTorch
2024. 7. 29. 21:05