pytorch 분류모델 학습이 한줄의 커맨드로 실행될 수 있는 도커 컨테이너를 구축한다.

git clone https://github.com/ProtossDragoon/PytorchTutorialDocker.git cd PytorchTutorialDocker docker-compose up pytorchtutorialdocker-train-and-eval-1 | /usr/local/lib/python3.8/site-packages/torchvision/io/image.py:13: UserWarning: Failed to load image Python extension: pytorchtutorialdocker-train-and-eval-1 | warn(f"Failed to load image Python extension: {e}") pytorchtutorialdocker-train-and-eval-1 | Files already downloaded and verified pytorchtutorialdocker-train-and-eval-1 | Files already downloaded and verified pytorchtutorialdocker-train-and-eval-1 | [1, 2000] loss: 2.204 pytorchtutorialdocker-train-and-eval-1 | [1, 4000] loss: 1.819 pytorchtutorialdocker-train-and-eval-1 | [1, 6000] loss: 1.649 pytorchtutorialdocker-train-and-eval-1 | [1, 8000] loss: 1.532 pytorchtutorialdocker-train-and-eval-1 | [1, 10000] loss: 1.468 pytorchtutorialdocker-train-and-eval-1 | [1, 12000] loss: 1.437 pytorchtutorialdocker-train-and-eval-1 | Finished Training pytorchtutorialdocker-train-and-eval-1 | Model saved pytorchtutorialdocker-train-and-eval-1 | GT: cat ship ship plane pytorchtutorialdocker-train-and-eval-1 | PRED: cat car car ship