Блог

Я новичок в мире тензорного потока, поэтому, пожалуйста, потерпите меня.

В настоящее время я использую MacBook Pro M1 и следую этим инструкциям по установке TensorFlow с помощью Conda: https://towardsdatascience.com/installing-tensorflow-on-the-m1-mac-410bb36b776

Весь код отлично работает в ноутбуке Jupyter, пока я не попытаюсь подогнать модель:

 initial_epochs = 5
history = model.fit(train_dataset, validation_data=validation_dataset, epochs=initial_epochs)
 

Однако после обучения для обучающих примеров 2175/2176 в 1-ю эпоху всегда появляется всплывающее окно, в котором отмечается, что ядро умерло и должно быть перезапущено. Для справки, вот выходные данные до появления этого всплывающего окна:

 Epoch 1/5
WARNING:tensorflow:AutoGraph could not transform <function Model.make_train_function.<locals>.train_function at 0x17934f040> and will run it as-is.
Please report this to the TensorFlow team. When filing the bug, set the verbosity to 10 (on Linux, `export AUTOGRAPH_VERBOSITY=10`) and attach the full output.
Cause: unsupported operand type(s) for -: 'NoneType' and 'int'
To silence this warning, decorate the function with @tf.autograph.experimental.do_not_convert
WARNING: AutoGraph could not transform <function Model.make_train_function.<locals>.train_function at 0x17934f040> and will run it as-is.
Please report this to the TensorFlow team. When filing the bug, set the verbosity to 10 (on Linux, `export AUTOGRAPH_VERBOSITY=10`) and attach the full output.
Cause: unsupported operand type(s) for -: 'NoneType' and 'int'
To silence this warning, decorate the function with @tf.autograph.experimental.do_not_convert
2175/2176 [============================>.] - ETA: 0s - loss: 1.6207 - accuracy: 0.5527
 

Я был бы очень признателен за любую помощь в этом, так как я уже некоторое время борюсь с этим!

Спасибо.

Вы можете проверить это с помощью ускорителя GPU, доступного в Runtime-Change runtime type-Hardware Accelerator . В зависимости от размера имеющегося у вас набора данных вы можете выбрать Colab Pro или TPU.

—ниже обновлено 07/15/21—

Мой предыдущий ответ касается Colab, извините, что я пропустил тему. Для ноутбука Jupyter должна быть панель, на которой возможен перезапуск мертвого ядра.