search

apple / tensorflow_macos

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

Segmentation fault at epoch n 3 #125

Open AndLouis opened 3 years ago

AndLouis commented 3 years ago

using this https://www.tensorflow.org/tutorials/images/transfer_learning

when mlcompute.set_mlc_device(device_name='any') throw a segmentation fault at epoch 3 when mlcompute.set_mlc_device(device_name='cpu') throw a segmentation fault at epoch 3 when mlcompute.set_mlc_device(device_name='gpu') it works but the log is wrong WARNING:tensorflow:Eager mode uses the CPU. Switching to the CPU. A little bit faster than the cpu but what can we deduce from 3 epochs.

anna-tikhonova commented 3 years ago

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