search

apple / tensorflow_macos

TensorFlow for macOS 11.0+ accelerated using Apple's ML Compute framework.
Other
3.67k stars 308 forks source link

Transfer learning and fine-tuning TensorFlow tutorial fails while running tf.keras... model.fit #84

Open mduaneh opened 3 years ago

mduaneh commented 3 years ago

I was trying out some simple tutorials from TensorFlow. Specifically this one:

When I get down to the first model.fit I get one, sometimes two epochs then the block finishes with no error. Yet the model fit returned no object.

This is the location in the tutorial: history = model.fit(train_dataset, epochs=initial_epochs, validation_data=validation_dataset)

This same code runs fine on Google Colab.

anna-tikhonova commented 3 years ago

Thank you very much for reporting this issue. We will investigate and report back.

owenschris commented 3 years ago

Was able to reproduce this. Getting through 1-2 epochs, same specs. Thanks, all.