Я получаю следующую ошибку при обучении модели:
AttributeError: у объекта 'BaseLogger' нет атрибута 'on_code_batch_begin'
Вот полный вывод:
C:\Users\Walther\PycharmProjects\ml_test\venv\Scripts\python.exe C:\Users\Walther\PycharmProjects\ml_test\venv\init.py
2020-05-02 16:02:08.495192: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cudart64_101.dll
2020-05-02 16:02:10.853680: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library nvcuda.dll
2020-05-02 16:02:10.878764: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1555] Found device 0 with properties:
pciBusID: 0000:01:00.0 name: GeForce GTX 750 Ti computeCapability: 5.0
coreClock: 1.1105GHz coreCount: 5 deviceMemorySize: 2.00GiB deviceMemoryBandwidth: 80.47GiB/s
2020-05-02 16:02:10.879026: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cudart64_101.dll
2020-05-02 16:02:10.882919: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cublas64_10.dll
2020-05-02 16:02:10.887333: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cufft64_10.dll
2020-05-02 16:02:10.888777: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library curand64_10.dll
2020-05-02 16:02:10.893496: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cusolver64_10.dll
2020-05-02 16:02:10.895941: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cusparse64_10.dll
2020-05-02 16:02:10.905611: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cudnn64_7.dll
2020-05-02 16:02:10.906466: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1697] Adding visible gpu devices: 0
2020-05-02 16:02:10.906827: I tensorflow/core/platform/cpu_feature_guard.cc:142] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2
2020-05-02 16:02:10.907750: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1555] Found device 0 with properties:
pciBusID: 0000:01:00.0 name: GeForce GTX 750 Ti computeCapability: 5.0
coreClock: 1.1105GHz coreCount: 5 deviceMemorySize: 2.00GiB deviceMemoryBandwidth: 80.47GiB/s
2020-05-02 16:02:10.908015: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cudart64_101.dll
2020-05-02 16:02:10.908151: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cublas64_10.dll
2020-05-02 16:02:10.908285: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cufft64_10.dll
2020-05-02 16:02:10.908416: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library curand64_10.dll
2020-05-02 16:02:10.908543: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cusolver64_10.dll
2020-05-02 16:02:10.908673: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cusparse64_10.dll
2020-05-02 16:02:10.908800: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cudnn64_7.dll
2020-05-02 16:02:10.909749: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1697] Adding visible gpu devices: 0
2020-05-02 16:02:12.127294: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1096] Device interconnect StreamExecutor with strength 1 edge matrix:
2020-05-02 16:02:12.127441: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1102] 0
2020-05-02 16:02:12.127526: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1115] 0: N
2020-05-02 16:02:12.128526: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1241] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 1367 MB memory) -> physical GPU (device: 0, name: GeForce GTX 750 Ti, pci bus id: 0000:01:00.0, compute capability: 5.0)
Train on 60000 samples, validate on 10000 samples
Epoch 1/5
2020-05-02 16:02:12.537046: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cublas64_10.dll
32/60000 [..............................] - ETA: 11:46 - loss: 128.2292 - accuracy: 0.0625
992/60000 [..............................] - ETA: 25s - loss: 340.5944 - accuracy: 0.0938
1952/60000 [..............................] - ETA: 14s - loss: 174.2262 - accuracy: 0.0989
2944/60000 [>.............................] - ETA: 10s - loss: 116.3000 - accuracy: 0.1005
3968/60000 [>.............................] - ETA: 8s - loss: 86.9575 - accuracy: 0.1031
4992/60000 [=>............................] - ETA: 7s - loss: 69.5958 - accuracy: 0.1070
6048/60000 [==>...........................] - ETA: 6s - loss: 57.8930 - accuracy: 0.1090
7072/60000 [==>...........................] - ETA: 5s - loss: 49.8437 - accuracy: 0.1072
8128/60000 [===>..........................] - ETA: 5s - loss: 43.6671 - accuracy: 0.1065
9120/60000 [===>..........................] - ETA: 4s - loss: 39.1678 - accuracy: 0.1063
10144/60000 [====>.........................] - ETA: 4s - loss: 35.4525 - accuracy: 0.1076
11168/60000 [====>.........................] - ETA: 4s - loss: 32.4129 - accuracy: 0.1083
12192/60000 [=====>........................] - ETA: 3s - loss: 29.8839 - accuracy: 0.1084
13184/60000 [=====>........................] - ETA: 3s - loss: 27.8086 - accuracy: 0.1082
14208/60000 [======>.......................] - ETA: 3s - loss: 25.9703 - accuracy: 0.1082
15232/60000 [======>.......................] - ETA: 3s - loss: 24.3792 - accuracy: 0.1087
16288/60000 [=======>......................] - ETA: 3s - loss: 22.9478 - accuracy: 0.1087
17280/60000 [=======>......................] - ETA: 3s - loss: 21.7625 - accuracy: 0.1104
18336/60000 [========>.....................] - ETA: 2s - loss: 20.6417 - accuracy: 0.1109
19360/60000 [========>.....................] - ETA: 2s - loss: 19.6717 - accuracy: 0.1107
20416/60000 [=========>....................] - ETA: 2s - loss: 18.7731 - accuracy: 0.1119
21440/60000 [=========>....................] - ETA: 2s - loss: 17.9864 - accuracy: 0.1121
22496/60000 [==========>...................] - ETA: 2s - loss: 17.2501 - accuracy: 0.1117
23520/60000 [==========>...................] - ETA: 2s - loss: 16.5992 - accuracy: 0.1118
24576/60000 [===========>..................] - ETA: 2s - loss: 15.9848 - accuracy: 0.1121
25600/60000 [===========>..................] - ETA: 2s - loss: 15.4428 - accuracy: 0.1122
26624/60000 [============>.................] - ETA: 2s - loss: 14.9373 - accuracy: 0.1129
27648/60000 [============>.................] - ETA: 2s - loss: 14.4693 - accuracy: 0.1133
28704/60000 [=============>................] - ETA: 1s - loss: 14.0216 - accuracy: 0.1136
29728/60000 [=============>................] - ETA: 1s - loss: 13.6180 - accuracy: 0.1130
30688/60000 [==============>...............] - ETA: 1s - loss: 13.2639 - accuracy: 0.1129
31712/60000 [==============>...............] - ETA: 1s - loss: 12.9100 - accuracy: 0.1127
32768/60000 [===============>..............] - ETA: 1s - loss: 12.5681 - accuracy: 0.1125
33792/60000 [===============>..............] - ETA: 1s - loss: 12.2570 - accuracy: 0.1121
34816/60000 [================>.............] - ETA: 1s - loss: 11.9642 - accuracy: 0.1120
35840/60000 [================>.............] - ETA: 1s - loss: 11.6881 - accuracy: 0.1119
36832/60000 [=================>............] - ETA: 1s - loss: 11.4353 - accuracy: 0.1120
37856/60000 [=================>............] - ETA: 1s - loss: 11.1882 - accuracy: 0.1125
38912/60000 [==================>...........] - ETA: 1s - loss: 10.9470 - accuracy: 0.1125
39968/60000 [==================>...........] - ETA: 1s - loss: 10.7186 - accuracy: 0.1123
41024/60000 [===================>..........] - ETA: 1s - loss: 10.5019 - accuracy: 0.1120
42048/60000 [====================>.........] - ETA: 1s - loss: 10.3022 - accuracy: 0.1123
43104/60000 [====================>.........] - ETA: 0s - loss: 10.1063 - accuracy: 0.1121
44128/60000 [=====================>........] - ETA: 0s - loss: 9.9251 - accuracy: 0.1122
45152/60000 [=====================>........] - ETA: 0s - loss: 9.7522 - accuracy: 0.1121
46176/60000 [======================>.......] - ETA: 0s - loss: 9.5870 - accuracy: 0.1123
47232/60000 [======================>.......] - ETA: 0s - loss: 9.4241 - accuracy: 0.1123
48288/60000 [=======================>......] - ETA: 0s - loss: 9.2684 - accuracy: 0.1123
49312/60000 [=======================>......] - ETA: 0s - loss: 9.1237 - accuracy: 0.1121
50304/60000 [========================>.....] - ETA: 0s - loss: 8.9892 - accuracy: 0.1119
51328/60000 [========================>.....] - ETA: 0s - loss: 8.8558 - accuracy: 0.1119
52384/60000 [=========================>....] - ETA: 0s - loss: 8.7237 - accuracy: 0.1118
53408/60000 [=========================>....] - ETA: 0s - loss: 8.6005 - accuracy: 0.1119
54336/60000 [==========================>...] - ETA: 0s - loss: 8.4930 - accuracy: 0.1115
55328/60000 [==========================>...] - ETA: 0s - loss: 8.3820 - accuracy: 0.1115
56320/60000 [===========================>..] - ETA: 0s - loss: 8.2749 - accuracy: 0.1112
57248/60000 [===========================>..] - ETA: 0s - loss: 8.1781 - accuracy: 0.1111
58144/60000 [============================>.] - ETA: 0s - loss: 8.0875 - accuracy: 0.1114
59072/60000 [============================>.] - ETA: 0s - loss: 7.9966 - accuracy: 0.1116
60000/60000 [==============================] - 3s 57us/sample - loss: 7.9085 - accuracy: 0.1117
Traceback (most recent call last):
File "C:\Users\Walther\PycharmProjects\ml_test\venv\init.py", line 17, in <module>
history = model.fit(X_train_full, y_train_full, epochs=5, validation_data=(X_test, y_test))
File "C:\Users\Walther\PycharmProjects\ml_test\venv\lib\site-packages\tensorflow_core\python\keras\engine\training.py", line 819, in fit
use_multiprocessing=use_multiprocessing)
File "C:\Users\Walther\PycharmProjects\ml_test\venv\lib\site-packages\tensorflow_core\python\keras\engine\training_v2.py", line 395, in fit
total_epochs=1)
File "C:\Users\Walther\PycharmProjects\ml_test\venv\lib\site-packages\tensorflow_core\python\keras\engine\training_v2.py", line 126, in run_one_epoch
step=step, mode=mode, size=current_batch_size) as batch_logs:
File "C:\Users\Walther\AppData\Local\Programs\Python\Python37\lib\contextlib.py", line 112, in __enter__
return next(self.gen)
File "C:\Users\Walther\PycharmProjects\ml_test\venv\lib\site-packages\tensorflow_core\python\keras\engine\training_v2.py", line 781, in on_batch
mode, 'begin', step, batch_logs)
File "C:\Users\Walther\PycharmProjects\ml_test\venv\lib\site-packages\tensorflow_core\python\keras\callbacks.py", line 238, in _call_batch_hook
batch_hook = getattr(callback, hook_name)
AttributeError: 'BaseLogger' object has no attribute 'on_code_batch_begin'
Process finished with exit code 1
Вот код:
from tensorflow import keras
(X_train_full, y_train_full), (X_test, y_test) = keras.datasets.mnist.load_data()
X_train_full = X_train_full.reshape(X_train_full.shape[0], -1)
X_test = X_test.reshape(X_test.shape[0], -1)
model = keras.models.Sequential()
model.add(keras.layers.InputLayer(input_shape=784))
model.add(keras.layers.Dense(50, activation='relu'))
model.add(keras.layers.Dense(20, activation='relu'))
model.add(keras.layers.Dense(10, activation='softmax'))
model.compile(loss='sparse_categorical_crossentropy',
optimizer='sgd',
metrics=['accuracy'])
history = model.fit(X_train_full, y_train_full, epochs=5, validation_data=(X_test, y_test))
print(history)
Кто-нибудь сейчас знает, почему возникает эта ошибка? Я также пробовал другую модель с разными слоями, но результат тот же.
Заранее спасибо.