PyTorch imagenet Transfer Learning
http://pytorch.org/tutorials/beginner/transfer_learning_tutorial.html
Ubuntu 미세먼지 팁들
sudo apt-get install lsb-release scrot wget http://github.com/downloads/djmelik/archey/archey-0.2.8.deb sudo dpkg -i archey-0.2.8.deb archey install
파이토치 인스털:
답글삭제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
파이토치 설치여부 확인:
답글삭제python에서 import torch 가 먹어줘야 ...
# 분류망 통과에 클래스 갯수 조정 !
답글삭제model_ft.fc = nn.Linear(num_ftrs, 3)