2017-03-16 5 views
1

https://github.com/fchollet/keras/issues/5818keras 2 - Jupyter 노트북의 fit_generator가 고장 났습니까?


어제 keras v2로 업데이트되었습니다.

내가 겪었던 모든 경고를 따르고 버전 1의 모든 코드를 새 API에 적용했습니다.

그러나 fit_generator 방법이 매우 이상한 문제가 있습니다. Model입니다. 이 장난감 예제를 사용하여

, 느릅 나무는 버전 1에서 완전히 잘 작동 :

jupyter 노트북의 출력은 노트북 충돌 할 때까지 알 수없는 기호를 인쇄, 아주 이상하다

from keras.models import Model 
from keras.layers import Input, Dense, Flatten 
from keras.optimizers import SGD 
from keras.losses import categorical_crossentropy 
from keras.preprocessing.image import ImageDataGenerator 

gen = ImageDataGenerator() 
train_batches = gen.flow_from_directory("D:/GitHub/Kaggle/redux/train/") 

inp = Input(shape=(256,256,3)) 
l1 = Flatten()(inp) 
out = Dense(2, activation="softmax")(l1) 

model = Model(inp, out) 

model.compile(loss=categorical_crossentropy, optimizer=SGD(lr=0.01)) 

model.fit_generator(train_batches, train_batches.samples // train_batches.batch_size) 
:

Epoch 1/1 
23/718 [..............................] - ETA: 522s - loss: 8.4146 

실행 터미널의 코드는 이상한 기호를 인쇄하지 않습니다.

+0

fit_generator의 두 번째 매개 변수에서 float division (/)을 사용하면 정수 나누기 (//)를 사용하면 어떻게됩니까? –

+0

어쩌면 어리석은 질문 일 수 있습니다.하지만 대답을 모르며 S.O가 질문을 올릴 곳입니다. 그런데 정수 나누기를 사용하면 같은 결과가 나타납니다. 수동으로 Integer를 전달하면 동일한 결과가 발생합니다. –

+0

"어리 석다"는 내 질문에 관한 것이지, 네 질문에 관한 것이 아니다. 나는 단지 오타 또는 사소한 부분을 없애려하고 있었다. 어쨌든, 제 자신의 네트워크에서 얻은 것부터 케라 2로 이동 한 후 카운터는 샘플 수가 아닌 일괄 처리 수를 표시하는 것으로 보입니다. –

답변

0

Jupyter 노트북에 오류 메시지가 있습니까? fit_generator에서 verbose = 0으로 설정할 수 있습니다. 그러면 진행률 표시 줄이 표시되지 않습니다.

관련 문제