Ядро перезапускается автоматически при использовании политики одного цикла с max lr 2e-05

Question

Как только я начну тренироваться с использованием политики одного цикла с max lr 2e-05 ...

Я получил,

Перезапуск ядра ...
[SpyderKernelApp] ВНИМАНИЕ | Нет такой связи: 3f9a86d8721c11eb94d81e00a2116544

Раньше я сталкивался с проблемой перезапуска ядра с помощью tensorflow, которая была исправлена ​​с помощью онлайн-справки. Но теперь возникла эта проблема. Использование на Macbook Pro M1

---

Вот мой полный код

      import os.path
import numpy as np
import tensorflow as tf
import ktrain
from ktrain import text

# Loading the IMDB dataset
dataset = tf.keras.utils.get_file(fname="aclImdb_v1.tar.gz",
                                  origin="http://ai.stanford.edu/~amaas/data/sentiment/aclImdb_v1.tar.gz",
                                  extract=True)
IMDB_DATADIR = os.path.join(os.path.dirname(dataset), 'aclImdb')

print(os.path.dirname(dataset))
print(IMDB_DATADIR)

# Creating the training and test sets
(x_train, y_train), (x_test, y_test), preproc = text.texts_from_folder(datadir=IMDB_DATADIR,
                                                                       classes=['pos','neg'],
                                                                       maxlen=500,
                                                                       train_test_names=['train','test'],
                                                                       preprocess_mode='bert')
# Building the BERT model
model = text.text_classifier(name='bert',
                             train_data=(x_train, y_train),
                             preproc=preproc)

# Training the BERT model
learner = ktrain.get_learner(model=model,
                             train_data=(x_train, y_train),
                             val_data=(x_test, y_test),
                             batch_size=6)

learner.fit_onecycle(lr=2e-5,
                     epochs=1)