2017년 11월 7일 화요일

Saving model in pytorch

torch.save(model_ft, "modelFT_60.pth")

댓글 없음:

댓글 쓰기

gpustat command

sudo apt install gpustat watch --color -n0.1 gpustat --color