Colab 속도 문제(CycleGAN-TensorFlow 학습)
vanhuyz/CycleGAN-TensorFlow 코드로 Colab 한 session으로 실행 했고,
대략 12시간째 학습 중이다.
너무 느리다는 이슈도 많고, 직접 체감하기에도 그런 것 같아서
xhujoy/CycleGAN-tensorflow 라는 다른 코드를 사용해봤다.
vanhuyz/CycleGAN-TensorFlow 코드
runtime설정: Python3 / GPU
지인 컴퓨터 사양이
CPU: i7-7700
GPU: 1050ti
RAM: 32GB
인데
github에 있는 cycleGAN-Tensorflow 코드로 학습이 하나 진행되는데 1초 정도 걸린다.
그런데 같은 코드를 내 컴퓨터에서 Colab으로 돌리면 거의 40초가 걸린다.
이번에는 추가로 Colab 창을 켜서 동시에 돌려보기로 했다.
즉, 기존에 돌리던 vanhuyz/CycleGAN-TensorFlow 코드가 계속 학습 되고 있고
동시에 xhujoy/CycleGAN-tensorflow 코드가 TPU 와 GPU 설정으로 하나씩 더 학습되고 있다.
xhujoy/CycleGAN-tensorflow 코드
runtime설정: Python3 / TPU
xhujoy/CycleGAN-tensorflow 코드
runtime설정: Python3 / GPU
xhujoy/CycleGAN-tensorflow 코드
지인 컴퓨터 학습 중인 모습
0 comments