PyTorch imagenet Transfer Learning

-
http://pytorch.org/tutorials/beginner/transfer_learning_tutorial.html

댓글

  1. 제갈식2017년 8월 15일 오후 11:10

    파이토치 인스털:

    pip install http://download.pytorch.org/whl/cu80/torch-0.1.12.post2-cp27-none-linux_x86_64.whl

    pip install torchvision


    운영체계 및 파이썬 버전별 설치:

    https://github.com/pytorch/pytorch

    답글삭제
  2. 제갈식2017년 8월 15일 오후 11:11

    파이토치 설치여부 확인:

    python에서 import torch 가 먹어줘야 ...

    답글삭제
  3. 제갈식2017년 8월 16일 오전 5:53

    # 분류망 통과에 클래스 갯수 조정 !
    model_ft.fc = nn.Linear(num_ftrs, 3)

    답글삭제

댓글 쓰기