PyTorch 모델 학습

PyTorch는 딥 러닝 네트워크를 빌드하기 위한 GPU 가속 텐서 계산 및 높은 수준의 기능을 제공하는 Python 패키지입니다.

MLflow PyTorch Notebook은 MNIST 필기 숫자 인식 데이터의 신경망에 적합하며 MLflow 서버에 실행 결과를 기록합니다. 학습 메트릭 및 가중치를 TensorFlow 이벤트 형식으로 로컬로 기록한 다음 MLflow 실행의 아티팩트 디렉터리에 업로드합니다. 마지막으로 TensorBoard를 시작하고 로컬로 로깅된 이벤트를 읽습니다.

준비가 되면 Azure Databricks로 모델 제공 서비스를 사용하여 모델을 배포할 수 있습니다.

MLflow PyTorch 모델 학습 Notebook

notebook 가져오기