2025-02-05 12:19:54.390955: I tensorflow/core/platform/cpu_feature_guard.cc:210] This TensorFlow binary is optimized to use available CPU instructions in performance-critical operations.
To enable the following instructions: AVX2 FMA, in other operations, rebuild TensorFlow with the appropriate compiler flags.
WARNING: All log messages before absl::InitializeLog() is called are written to STDERR
I0000 00:00:1738729197.544741    4373 gpu_device.cc:2019] Created device /job:localhost/replica:0/task:0/device:GPU:0 with 3315 MB memory:  -> device: 0, name: NVIDIA GeForce GTX 1050 Ti, pci bus id: 0000:29:00.0, compute capability: 6.1
Model: "functional_4"
┏━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━┓
┃ Layer (type)                    ┃ Output Shape           ┃       Param # ┃
┡━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━┩
│ input_layer (InputLayer)        │ (None, None, 128)      │             0 │
├─────────────────────────────────┼────────────────────────┼───────────────┤
│ encoder (Encoder)               │ (None, None, 128)      │       662,528 │
├─────────────────────────────────┼────────────────────────┼───────────────┤
│ out_pretraining (Dense)         │ (None, None, 128)      │        16,512 │
└─────────────────────────────────┴────────────────────────┴───────────────┘
 Total params: 679,040 (2.59 MB)
 Trainable params: 679,040 (2.59 MB)
 Non-trainable params: 0 (0.00 B)