Skip to content

Instantly share code, notes, and snippets.

@ranpelta

ranpelta/step 8 fit the model and plot losses.py

Created July 18, 2020 07:57
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save ranpelta/e601a642b672a22c227d51d40b00da41 to your computer and use it in GitHub Desktop.
Save ranpelta/e601a642b672a22c227d51d40b00da41 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
step 8 fit the model and plot losses.py
model.fit_generator(generator,epochs=5)
>>>
Epoch 1/5
903/903 [==============================] - 14s 15ms/step - loss: 0.0311
Epoch 2/5
903/903 [==============================] - 14s 15ms/step - loss: 0.0189
Epoch 3/5
903/903 [==============================] - 14s 16ms/step - loss: 0.0170
Epoch 4/5
903/903 [==============================] - 15s 16ms/step - loss: 0.0161
Epoch 5/5
903/903 [==============================] - 15s 16ms/step - loss: 0.0152
>>>
loss_per_epoch = model.history.history['loss']
plt.plot(range(len(loss_per_epoch)),loss_per_epoch);
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment