2022-07-04 11:22:03.313810: I tensorflow/core/platform/cpu_feature_guard.cc:142] This TensorFlow binary is optimized with oneAPI Deep Neural Network Library (oneDNN) to use the following CPU instructions in performance-critical operations:  SSE4.2 AVX AVX2 FMA
To enable them in other operations, rebuild TensorFlow with the appropriate compiler flags.
Metal device set to: AMD Radeon PRO W6800

systemMemory: 64.00 GB
maxCacheSize: 14.99 GB

2022-07-04 11:22:03.314356: I tensorflow/core/common_runtime/pluggable_device/pluggable_device_factory.cc:305] Could not identify NUMA node of platform GPU ID 0, defaulting to 0. Your kernel may not have been built with NUMA support.
2022-07-04 11:22:03.314618: I tensorflow/core/common_runtime/pluggable_device/pluggable_device_factory.cc:271] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 0 MB memory) -> physical PluggableDevice (device: 0, name: METAL, pci bus id: <undefined>)
Epoch 1/20
2022-07-04 11:22:08.308800: I tensorflow/compiler/mlir/mlir_graph_optimization_pass.cc:185] None of the MLIR Optimization Passes are enabled (registered 2)
2022-07-04 11:22:08.309831: I tensorflow/core/grappler/optimizers/custom_graph_optimizer_registry.cc:112] Plugin optimizer for device_type GPU is enabled.
20/20 [==============================] - 24s 1s/step - loss: 2.2355
Epoch 2/20
20/20 [==============================] - 23s 1s/step - loss: 2.1982
Epoch 3/20
20/20 [==============================] - 23s 1s/step - loss: 2.1975
Epoch 4/20
20/20 [==============================] - 23s 1s/step - loss: 2.1974
Epoch 5/20
20/20 [==============================] - 23s 1s/step - loss: 2.1974
Epoch 6/20
20/20 [==============================] - 23s 1s/step - loss: 2.1974
Epoch 7/20
20/20 [==============================] - 23s 1s/step - loss: 2.1974
Epoch 8/20
20/20 [==============================] - 23s 1s/step - loss: 2.1974
Epoch 9/20
20/20 [==============================] - 23s 1s/step - loss: 2.1974
Epoch 10/20
20/20 [==============================] - 23s 1s/step - loss: 2.1973
Epoch 11/20
20/20 [==============================] - 23s 1s/step - loss: 2.1973
Epoch 12/20
20/20 [==============================] - 23s 1s/step - loss: 2.1972
Epoch 13/20
20/20 [==============================] - 23s 1s/step - loss: 2.1972
Epoch 14/20
20/20 [==============================] - 23s 1s/step - loss: 2.1972
Epoch 15/20
20/20 [==============================] - 23s 1s/step - loss: 2.1971
Epoch 16/20
20/20 [==============================] - 23s 1s/step - loss: 2.1971
Epoch 17/20
20/20 [==============================] - 23s 1s/step - loss: 2.1970
Epoch 18/20
20/20 [==============================] - 23s 1s/step - loss: 2.1970
Epoch 19/20
20/20 [==============================] - 23s 1s/step - loss: 2.1970
Epoch 20/20
20/20 [==============================] - 23s 1s/step - loss: 2.1969