딥러닝프레임워크1 torch의 model summary에서는 cuda가 필요한가요? 이 글은 이곳을 번역한 글이며 추가로 제 경험도 들어있습니다. 케라스에는 아주 편리하게 model.summary() 하면 레이어에 따른 파라미터 갯수가 쭉쭉 나온다. 파이토치는 따로 import 시켜야 그걸 할 수 있는데 아래 처럼 사용하면 된다. from torchsummary import summary summary(vgg, (3, 224, 224)) 그러면 바로 아래 사진처럼 쭉쭉 나와야한다. 내가 참고한 글부터 이야기하자면 이렇다. summary를 사용하면 아래와 같은 에러가 뜬다고 한다. RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same "input타입은 to.. 2021. 1. 5. 이전 1 다음
