개요
케라스에서 Multi GPU를 사용하는 코드를 정리해둔다.
코드
패키지를 아래와 같이 로드한다.
model = multi_gpu_model(model, gpus=4)
를 추가해준다.
model = cnn_model()
model = multi_gpu_model(model, gpus=4)
model.compile(loss='categorical_crossentropy', optimizer='adam', metrics=['accuracy'])
model.fit(train_x, train_y, validation_data=(test_x, test_y), nb_epoch=20, batch_size=32, verbose=1)