TensorFlow 애드온 손실: TripletSemiHardLoss

TensorFlow.org에서 보기 Google Colab에서 실행하기 GitHub에서 소스 보기 노트북 다운로드하기

개요

이 노트북은 TensorFlow 애드온에서 TripletSemiHardLoss 함수를 사용하는 방법을 보여줍니다.

리소스:

TripletLoss

FaceNet 논문에 처음 소개된 TripletLoss는 신경망을 훈련하여 같은 클래스의 특성을 밀접하게 포함하면서 서로 다른 클래스의 임베딩 간의 거리를 최대화합니다. 이를 위해 하나의 음수 샘플과 하나의 양수 샘플과 함께 앵커가 선택됩니다. fig3

손실 함수는 Euclidean 거리 함수로 설명됩니다.

function

A는 앵커 입력이고, P는 양의 샘플 입력이고, N은 음의 샘플 입력이며, 알파는 트리플릿이 너무 "쉬워지는" 시점을 지정하기 위해 사용하는 한계입니다. .

SemiHard 온라인 학습

이 논문에서 볼 수 있듯이 가장 좋은 결과는 "Semi-Hard"로 알려진 트리플릿에서 얻습니다. 트리플릿은 음이 양보다 앵커에서 더 멀리 있는 트리플릿으로 정의되지만, 여전히 양의 손실을 생성합니다. 이러한 트리플릿을 효율적으로 찾기 위해 온라인 학습을 활용하고 각 배치에서 Semi-Hard 예제를 통해서만 훈련합니다.

설정

pip install -q -U tensorflow-addons
import io
import numpy as np
import tensorflow as tf
import tensorflow_addons as tfa
import tensorflow_datasets as tfds

데이터 준비하기

def _normalize_img(img, label):
    img = tf.cast(img, tf.float32) / 255.
    return (img, label)

train_dataset, test_dataset = tfds.load(name="mnist", split=['train', 'test'], as_supervised=True)

# Build your input pipelines
train_dataset = train_dataset.shuffle(1024).batch(32)
train_dataset = train_dataset.map(_normalize_img)

test_dataset = test_dataset.batch(32)
test_dataset = test_dataset.map(_normalize_img)
Downloading and preparing dataset mnist/3.0.1 (download: 11.06 MiB, generated: 21.00 MiB, total: 32.06 MiB) to /home/kbuilder/tensorflow_datasets/mnist/3.0.1...
WARNING:absl:Dataset mnist is hosted on GCS. It will automatically be downloaded to your
local data directory. If you'd instead prefer to read directly from our public
GCS bucket (recommended if you're running on GCP), you can instead pass
`try_gcs=True` to `tfds.load` or set `data_dir=gs://tfds-data/datasets`.
Dataset mnist downloaded and prepared to /home/kbuilder/tensorflow_datasets/mnist/3.0.1. Subsequent calls will reuse this data.

모델 빌드하기

fig2

model = tf.keras.Sequential([
    tf.keras.layers.Conv2D(filters=64, kernel_size=2, padding='same', activation='relu', input_shape=(28,28,1)),
    tf.keras.layers.MaxPooling2D(pool_size=2),
    tf.keras.layers.Dropout(0.3),
    tf.keras.layers.Conv2D(filters=32, kernel_size=2, padding='same', activation='relu'),
    tf.keras.layers.MaxPooling2D(pool_size=2),
    tf.keras.layers.Dropout(0.3),
    tf.keras.layers.Flatten(),
    tf.keras.layers.Dense(256, activation=None), # No activation on final dense layer
    tf.keras.layers.Lambda(lambda x: tf.math.l2_normalize(x, axis=1)) # L2 normalize embeddings

])

훈련 및 평가하기

# Compile the model
model.compile(
    optimizer=tf.keras.optimizers.Adam(0.001),
    loss=tfa.losses.TripletSemiHardLoss())
# Train the network
history = model.fit(
    train_dataset,
    epochs=5)
Epoch 1/5
1/1875 [..............................] - ETA: 1s - loss: 0.9853
12/1875 [..............................] - ETA: 7s - loss: 0.9757
24/1875 [..............................] - ETA: 7s - loss: 0.9612
36/1875 [..............................] - ETA: 7s - loss: 0.9524
49/1875 [..............................] - ETA: 7s - loss: 0.9398
62/1875 [..............................] - ETA: 7s - loss: 0.9289
75/1875 [>.............................] - ETA: 7s - loss: 0.9212
88/1875 [>.............................] - ETA: 7s - loss: 0.9113
100/1875 [>.............................] - ETA: 7s - loss: 0.9048
112/1875 [>.............................] - ETA: 7s - loss: 0.8993
124/1875 [>.............................] - ETA: 7s - loss: 0.8895
136/1875 [=>............................] - ETA: 7s - loss: 0.8802
148/1875 [=>............................] - ETA: 7s - loss: 0.8751
160/1875 [=>............................] - ETA: 7s - loss: 0.8690
172/1875 [=>............................] - ETA: 7s - loss: 0.8639
184/1875 [=>............................] - ETA: 7s - loss: 0.8566
197/1875 [==>...........................] - ETA: 7s - loss: 0.8496
209/1875 [==>...........................] - ETA: 7s - loss: 0.8430
221/1875 [==>...........................] - ETA: 7s - loss: 0.8360
233/1875 [==>...........................] - ETA: 6s - loss: 0.8298
245/1875 [==>...........................] - ETA: 6s - loss: 0.8241
256/1875 [===>..........................] - ETA: 6s - loss: 0.8200
268/1875 [===>..........................] - ETA: 6s - loss: 0.8138
281/1875 [===>..........................] - ETA: 6s - loss: 0.8083
294/1875 [===>..........................] - ETA: 6s - loss: 0.8034
306/1875 [===>..........................] - ETA: 6s - loss: 0.7998
318/1875 [====>.........................] - ETA: 6s - loss: 0.7973
331/1875 [====>.........................] - ETA: 6s - loss: 0.7922
344/1875 [====>.........................] - ETA: 6s - loss: 0.7893
357/1875 [====>.........................] - ETA: 6s - loss: 0.7853
370/1875 [====>.........................] - ETA: 6s - loss: 0.7807
384/1875 [=====>........................] - ETA: 6s - loss: 0.7779
398/1875 [=====>........................] - ETA: 6s - loss: 0.7736
411/1875 [=====>........................] - ETA: 6s - loss: 0.7710
424/1875 [=====>........................] - ETA: 6s - loss: 0.7677
437/1875 [=====>........................] - ETA: 6s - loss: 0.7635
450/1875 [======>.......................] - ETA: 5s - loss: 0.7592
463/1875 [======>.......................] - ETA: 5s - loss: 0.7552
477/1875 [======>.......................] - ETA: 5s - loss: 0.7523
491/1875 [======>.......................] - ETA: 5s - loss: 0.7493
505/1875 [=======>......................] - ETA: 5s - loss: 0.7449
519/1875 [=======>......................] - ETA: 5s - loss: 0.7413
533/1875 [=======>......................] - ETA: 5s - loss: 0.7375
547/1875 [=======>......................] - ETA: 5s - loss: 0.7346
560/1875 [=======>......................] - ETA: 5s - loss: 0.7327
574/1875 [========>.....................] - ETA: 5s - loss: 0.7292
588/1875 [========>.....................] - ETA: 5s - loss: 0.7261
602/1875 [========>.....................] - ETA: 5s - loss: 0.7220
616/1875 [========>.....................] - ETA: 5s - loss: 0.7189
629/1875 [=========>....................] - ETA: 5s - loss: 0.7161
642/1875 [=========>....................] - ETA: 5s - loss: 0.7128
656/1875 [=========>....................] - ETA: 4s - loss: 0.7090
670/1875 [=========>....................] - ETA: 4s - loss: 0.7059
684/1875 [=========>....................] - ETA: 4s - loss: 0.7029
698/1875 [==========>...................] - ETA: 4s - loss: 0.7003
711/1875 [==========>...................] - ETA: 4s - loss: 0.6987
724/1875 [==========>...................] - ETA: 4s - loss: 0.6956
737/1875 [==========>...................] - ETA: 4s - loss: 0.6932
751/1875 [===========>..................] - ETA: 4s - loss: 0.6909
763/1875 [===========>..................] - ETA: 4s - loss: 0.6904
776/1875 [===========>..................] - ETA: 4s - loss: 0.6881
789/1875 [===========>..................] - ETA: 4s - loss: 0.6859
802/1875 [===========>..................] - ETA: 4s - loss: 0.6836
815/1875 [============>.................] - ETA: 4s - loss: 0.6815
828/1875 [============>.................] - ETA: 4s - loss: 0.6798
841/1875 [============>.................] - ETA: 4s - loss: 0.6784
855/1875 [============>.................] - ETA: 4s - loss: 0.6761
869/1875 [============>.................] - ETA: 4s - loss: 0.6740
883/1875 [=============>................] - ETA: 3s - loss: 0.6717
896/1875 [=============>................] - ETA: 3s - loss: 0.6698
910/1875 [=============>................] - ETA: 3s - loss: 0.6674
923/1875 [=============>................] - ETA: 3s - loss: 0.6658
936/1875 [=============>................] - ETA: 3s - loss: 0.6642
949/1875 [==============>...............] - ETA: 3s - loss: 0.6624
963/1875 [==============>...............] - ETA: 3s - loss: 0.6611
977/1875 [==============>...............] - ETA: 3s - loss: 0.6593
991/1875 [==============>...............] - ETA: 3s - loss: 0.6581
1005/1875 [===============>..............] - ETA: 3s - loss: 0.6564
1019/1875 [===============>..............] - ETA: 3s - loss: 0.6549
1032/1875 [===============>..............] - ETA: 3s - loss: 0.6533
1045/1875 [===============>..............] - ETA: 3s - loss: 0.6520
1058/1875 [===============>..............] - ETA: 3s - loss: 0.6502
1071/1875 [================>.............] - ETA: 3s - loss: 0.6490
1085/1875 [================>.............] - ETA: 3s - loss: 0.6477
1098/1875 [================>.............] - ETA: 3s - loss: 0.6464
1111/1875 [================>.............] - ETA: 3s - loss: 0.6454
1125/1875 [=================>............] - ETA: 2s - loss: 0.6445
1138/1875 [=================>............] - ETA: 2s - loss: 0.6435
1151/1875 [=================>............] - ETA: 2s - loss: 0.6421
1163/1875 [=================>............] - ETA: 2s - loss: 0.6409
1176/1875 [=================>............] - ETA: 2s - loss: 0.6391
1190/1875 [==================>...........] - ETA: 2s - loss: 0.6379
1203/1875 [==================>...........] - ETA: 2s - loss: 0.6368
1217/1875 [==================>...........] - ETA: 2s - loss: 0.6350
1230/1875 [==================>...........] - ETA: 2s - loss: 0.6340
1243/1875 [==================>...........] - ETA: 2s - loss: 0.6330
1257/1875 [===================>..........] - ETA: 2s - loss: 0.6316
1270/1875 [===================>..........] - ETA: 2s - loss: 0.6307
1283/1875 [===================>..........] - ETA: 2s - loss: 0.6295
1296/1875 [===================>..........] - ETA: 2s - loss: 0.6280
1309/1875 [===================>..........] - ETA: 2s - loss: 0.6271
1322/1875 [====================>.........] - ETA: 2s - loss: 0.6258
1336/1875 [====================>.........] - ETA: 2s - loss: 0.6249
1350/1875 [====================>.........] - ETA: 2s - loss: 0.6235
1364/1875 [====================>.........] - ETA: 2s - loss: 0.6226
1378/1875 [=====================>........] - ETA: 1s - loss: 0.6216
1392/1875 [=====================>........] - ETA: 1s - loss: 0.6204
1406/1875 [=====================>........] - ETA: 1s - loss: 0.6190
1420/1875 [=====================>........] - ETA: 1s - loss: 0.6182
1432/1875 [=====================>........] - ETA: 1s - loss: 0.6171
1445/1875 [======================>.......] - ETA: 1s - loss: 0.6163
1459/1875 [======================>.......] - ETA: 1s - loss: 0.6152
1473/1875 [======================>.......] - ETA: 1s - loss: 0.6145
1487/1875 [======================>.......] - ETA: 1s - loss: 0.6135
1501/1875 [=======================>......] - ETA: 1s - loss: 0.6124
1515/1875 [=======================>......] - ETA: 1s - loss: 0.6113
1528/1875 [=======================>......] - ETA: 1s - loss: 0.6105
1542/1875 [=======================>......] - ETA: 1s - loss: 0.6099
1556/1875 [=======================>......] - ETA: 1s - loss: 0.6090
1570/1875 [========================>.....] - ETA: 1s - loss: 0.6080
1584/1875 [========================>.....] - ETA: 1s - loss: 0.6068
1598/1875 [========================>.....] - ETA: 1s - loss: 0.6062
1611/1875 [========================>.....] - ETA: 1s - loss: 0.6051
1624/1875 [========================>.....] - ETA: 0s - loss: 0.6043
1638/1875 [=========================>....] - ETA: 0s - loss: 0.6034
1651/1875 [=========================>....] - ETA: 0s - loss: 0.6025
1664/1875 [=========================>....] - ETA: 0s - loss: 0.6014
1678/1875 [=========================>....] - ETA: 0s - loss: 0.6001
1691/1875 [==========================>...] - ETA: 0s - loss: 0.5995
1705/1875 [==========================>...] - ETA: 0s - loss: 0.5985
1719/1875 [==========================>...] - ETA: 0s - loss: 0.5975
1733/1875 [==========================>...] - ETA: 0s - loss: 0.5963
1747/1875 [==========================>...] - ETA: 0s - loss: 0.5956
1761/1875 [===========================>..] - ETA: 0s - loss: 0.5948
1775/1875 [===========================>..] - ETA: 0s - loss: 0.5940
1789/1875 [===========================>..] - ETA: 0s - loss: 0.5933
1803/1875 [===========================>..] - ETA: 0s - loss: 0.5926
1817/1875 [============================>.] - ETA: 0s - loss: 0.5916
1831/1875 [============================>.] - ETA: 0s - loss: 0.5908
1844/1875 [============================>.] - ETA: 0s - loss: 0.5899
1859/1875 [============================>.] - ETA: 0s - loss: 0.5892
1875/1875 [==============================] - ETA: 0s - loss: 0.5884
1875/1875 [==============================] - 7s 4ms/step - loss: 0.5884
Epoch 2/5
1/1875 [..............................] - ETA: 0s - loss: 0.6143
16/1875 [..............................] - ETA: 6s - loss: 0.5018
30/1875 [..............................] - ETA: 6s - loss: 0.4816
45/1875 [..............................] - ETA: 6s - loss: 0.4814
61/1875 [..............................] - ETA: 6s - loss: 0.4666
76/1875 [>.............................] - ETA: 6s - loss: 0.4742
91/1875 [>.............................] - ETA: 6s - loss: 0.4771
106/1875 [>.............................] - ETA: 6s - loss: 0.4830
121/1875 [>.............................] - ETA: 6s - loss: 0.4836
135/1875 [=>............................] - ETA: 6s - loss: 0.4855
150/1875 [=>............................] - ETA: 5s - loss: 0.4831
165/1875 [=>............................] - ETA: 5s - loss: 0.4824
180/1875 [=>............................] - ETA: 5s - loss: 0.4824
195/1875 [==>...........................] - ETA: 5s - loss: 0.4808
210/1875 [==>...........................] - ETA: 5s - loss: 0.4779
225/1875 [==>...........................] - ETA: 5s - loss: 0.4774
240/1875 [==>...........................] - ETA: 5s - loss: 0.4800
255/1875 [===>..........................] - ETA: 5s - loss: 0.4779
270/1875 [===>..........................] - ETA: 5s - loss: 0.4780
285/1875 [===>..........................] - ETA: 5s - loss: 0.4788
300/1875 [===>..........................] - ETA: 5s - loss: 0.4774
315/1875 [====>.........................] - ETA: 5s - loss: 0.4767
330/1875 [====>.........................] - ETA: 5s - loss: 0.4751
345/1875 [====>.........................] - ETA: 5s - loss: 0.4750
360/1875 [====>.........................] - ETA: 5s - loss: 0.4759
375/1875 [=====>........................] - ETA: 5s - loss: 0.4754
390/1875 [=====>........................] - ETA: 5s - loss: 0.4759
405/1875 [=====>........................] - ETA: 5s - loss: 0.4753
419/1875 [=====>........................] - ETA: 5s - loss: 0.4760
434/1875 [=====>........................] - ETA: 4s - loss: 0.4767
449/1875 [======>.......................] - ETA: 4s - loss: 0.4763
464/1875 [======>.......................] - ETA: 4s - loss: 0.4754
479/1875 [======>.......................] - ETA: 4s - loss: 0.4741
493/1875 [======>.......................] - ETA: 4s - loss: 0.4732
508/1875 [=======>......................] - ETA: 4s - loss: 0.4718
523/1875 [=======>......................] - ETA: 4s - loss: 0.4722
538/1875 [=======>......................] - ETA: 4s - loss: 0.4706
553/1875 [=======>......................] - ETA: 4s - loss: 0.4719
568/1875 [========>.....................] - ETA: 4s - loss: 0.4720
583/1875 [========>.....................] - ETA: 4s - loss: 0.4716
598/1875 [========>.....................] - ETA: 4s - loss: 0.4719
613/1875 [========>.....................] - ETA: 4s - loss: 0.4699
628/1875 [=========>....................] - ETA: 4s - loss: 0.4689
641/1875 [=========>....................] - ETA: 4s - loss: 0.4674
656/1875 [=========>....................] - ETA: 4s - loss: 0.4664
670/1875 [=========>....................] - ETA: 4s - loss: 0.4653
685/1875 [=========>....................] - ETA: 4s - loss: 0.4654
699/1875 [==========>...................] - ETA: 4s - loss: 0.4659
713/1875 [==========>...................] - ETA: 4s - loss: 0.4659
728/1875 [==========>...................] - ETA: 3s - loss: 0.4652
743/1875 [==========>...................] - ETA: 3s - loss: 0.4655
758/1875 [===========>..................] - ETA: 3s - loss: 0.4653
772/1875 [===========>..................] - ETA: 3s - loss: 0.4652
787/1875 [===========>..................] - ETA: 3s - loss: 0.4644
802/1875 [===========>..................] - ETA: 3s - loss: 0.4645
817/1875 [============>.................] - ETA: 3s - loss: 0.4644
832/1875 [============>.................] - ETA: 3s - loss: 0.4645
847/1875 [============>.................] - ETA: 3s - loss: 0.4642
861/1875 [============>.................] - ETA: 3s - loss: 0.4635
876/1875 [=============>................] - ETA: 3s - loss: 0.4628
891/1875 [=============>................] - ETA: 3s - loss: 0.4628
906/1875 [=============>................] - ETA: 3s - loss: 0.4626
921/1875 [=============>................] - ETA: 3s - loss: 0.4625
937/1875 [=============>................] - ETA: 3s - loss: 0.4624
952/1875 [==============>...............] - ETA: 3s - loss: 0.4627
967/1875 [==============>...............] - ETA: 3s - loss: 0.4628
981/1875 [==============>...............] - ETA: 3s - loss: 0.4628
995/1875 [==============>...............] - ETA: 3s - loss: 0.4624
1010/1875 [===============>..............] - ETA: 3s - loss: 0.4620
1025/1875 [===============>..............] - ETA: 2s - loss: 0.4624
1039/1875 [===============>..............] - ETA: 2s - loss: 0.4624
1054/1875 [===============>..............] - ETA: 2s - loss: 0.4619
1069/1875 [================>.............] - ETA: 2s - loss: 0.4621
1084/1875 [================>.............] - ETA: 2s - loss: 0.4622
1099/1875 [================>.............] - ETA: 2s - loss: 0.4621
1114/1875 [================>.............] - ETA: 2s - loss: 0.4617
1129/1875 [=================>............] - ETA: 2s - loss: 0.4616
1143/1875 [=================>............] - ETA: 2s - loss: 0.4615
1158/1875 [=================>............] - ETA: 2s - loss: 0.4615
1172/1875 [=================>............] - ETA: 2s - loss: 0.4614
1187/1875 [=================>............] - ETA: 2s - loss: 0.4611
1201/1875 [==================>...........] - ETA: 2s - loss: 0.4606
1215/1875 [==================>...........] - ETA: 2s - loss: 0.4605
1230/1875 [==================>...........] - ETA: 2s - loss: 0.4609
1245/1875 [==================>...........] - ETA: 2s - loss: 0.4606
1260/1875 [===================>..........] - ETA: 2s - loss: 0.4604
1274/1875 [===================>..........] - ETA: 2s - loss: 0.4609
1287/1875 [===================>..........] - ETA: 2s - loss: 0.4605
1302/1875 [===================>..........] - ETA: 2s - loss: 0.4603
1317/1875 [====================>.........] - ETA: 1s - loss: 0.4595
1332/1875 [====================>.........] - ETA: 1s - loss: 0.4590
1346/1875 [====================>.........] - ETA: 1s - loss: 0.4590
1360/1875 [====================>.........] - ETA: 1s - loss: 0.4588
1375/1875 [=====================>........] - ETA: 1s - loss: 0.4584
1390/1875 [=====================>........] - ETA: 1s - loss: 0.4580
1405/1875 [=====================>........] - ETA: 1s - loss: 0.4574
1420/1875 [=====================>........] - ETA: 1s - loss: 0.4574
1435/1875 [=====================>........] - ETA: 1s - loss: 0.4571
1450/1875 [======================>.......] - ETA: 1s - loss: 0.4575
1465/1875 [======================>.......] - ETA: 1s - loss: 0.4574
1480/1875 [======================>.......] - ETA: 1s - loss: 0.4571
1495/1875 [======================>.......] - ETA: 1s - loss: 0.4569
1510/1875 [=======================>......] - ETA: 1s - loss: 0.4575
1525/1875 [=======================>......] - ETA: 1s - loss: 0.4573
1539/1875 [=======================>......] - ETA: 1s - loss: 0.4570
1554/1875 [=======================>......] - ETA: 1s - loss: 0.4571
1569/1875 [========================>.....] - ETA: 1s - loss: 0.4571
1584/1875 [========================>.....] - ETA: 1s - loss: 0.4569
1599/1875 [========================>.....] - ETA: 0s - loss: 0.4566
1614/1875 [========================>.....] - ETA: 0s - loss: 0.4570
1629/1875 [=========================>....] - ETA: 0s - loss: 0.4568
1644/1875 [=========================>....] - ETA: 0s - loss: 0.4563
1659/1875 [=========================>....] - ETA: 0s - loss: 0.4561
1674/1875 [=========================>....] - ETA: 0s - loss: 0.4563
1689/1875 [==========================>...] - ETA: 0s - loss: 0.4559
1704/1875 [==========================>...] - ETA: 0s - loss: 0.4560
1719/1875 [==========================>...] - ETA: 0s - loss: 0.4553
1734/1875 [==========================>...] - ETA: 0s - loss: 0.4550
1748/1875 [==========================>...] - ETA: 0s - loss: 0.4549
1763/1875 [===========================>..] - ETA: 0s - loss: 0.4547
1778/1875 [===========================>..] - ETA: 0s - loss: 0.4545
1792/1875 [===========================>..] - ETA: 0s - loss: 0.4547
1807/1875 [===========================>..] - ETA: 0s - loss: 0.4544
1822/1875 [============================>.] - ETA: 0s - loss: 0.4543
1837/1875 [============================>.] - ETA: 0s - loss: 0.4539
1850/1875 [============================>.] - ETA: 0s - loss: 0.4535
1864/1875 [============================>.] - ETA: 0s - loss: 0.4534
1875/1875 [==============================] - 7s 3ms/step - loss: 0.4532
Epoch 3/5

   1/1875 [..............................] - ETA: 0s - loss: 0.4001
16/1875 [..............................] - ETA: 6s - loss: 0.3656
31/1875 [..............................] - ETA: 6s - loss: 0.3731
45/1875 [..............................] - ETA: 6s - loss: 0.3922
59/1875 [..............................] - ETA: 6s - loss: 0.4097
73/1875 [>.............................] - ETA: 6s - loss: 0.4188
88/1875 [>.............................] - ETA: 6s - loss: 0.4315
103/1875 [>.............................] - ETA: 6s - loss: 0.4304
118/1875 [>.............................] - ETA: 6s - loss: 0.4320
133/1875 [=>............................] - ETA: 6s - loss: 0.4299
148/1875 [=>............................] - ETA: 5s - loss: 0.4261
162/1875 [=>............................] - ETA: 5s - loss: 0.4274
177/1875 [=>............................] - ETA: 5s - loss: 0.4282
192/1875 [==>...........................] - ETA: 5s - loss: 0.4287
207/1875 [==>...........................] - ETA: 5s - loss: 0.4247
222/1875 [==>...........................] - ETA: 5s - loss: 0.4242
237/1875 [==>...........................] - ETA: 5s - loss: 0.4252
251/1875 [===>..........................] - ETA: 5s - loss: 0.4277
266/1875 [===>..........................] - ETA: 5s - loss: 0.4275
281/1875 [===>..........................] - ETA: 5s - loss: 0.4275
296/1875 [===>..........................] - ETA: 5s - loss: 0.4257
311/1875 [===>..........................] - ETA: 5s - loss: 0.4256
325/1875 [====>.........................] - ETA: 5s - loss: 0.4264
339/1875 [====>.........................] - ETA: 5s - loss: 0.4264
354/1875 [====>.........................] - ETA: 5s - loss: 0.4273
368/1875 [====>.........................] - ETA: 5s - loss: 0.4267
383/1875 [=====>........................] - ETA: 5s - loss: 0.4257
398/1875 [=====>........................] - ETA: 5s - loss: 0.4248
413/1875 [=====>........................] - ETA: 5s - loss: 0.4242
428/1875 [=====>........................] - ETA: 5s - loss: 0.4247
443/1875 [======>.......................] - ETA: 5s - loss: 0.4259
458/1875 [======>.......................] - ETA: 4s - loss: 0.4273
472/1875 [======>.......................] - ETA: 4s - loss: 0.4263
487/1875 [======>.......................] - ETA: 4s - loss: 0.4260
502/1875 [=======>......................] - ETA: 4s - loss: 0.4256
516/1875 [=======>......................] - ETA: 4s - loss: 0.4246
531/1875 [=======>......................] - ETA: 4s - loss: 0.4251
546/1875 [=======>......................] - ETA: 4s - loss: 0.4257
560/1875 [=======>......................] - ETA: 4s - loss: 0.4253
575/1875 [========>.....................] - ETA: 4s - loss: 0.4250
590/1875 [========>.....................] - ETA: 4s - loss: 0.4245
605/1875 [========>.....................] - ETA: 4s - loss: 0.4244
619/1875 [========>.....................] - ETA: 4s - loss: 0.4238
633/1875 [=========>....................] - ETA: 4s - loss: 0.4232
648/1875 [=========>....................] - ETA: 4s - loss: 0.4218
663/1875 [=========>....................] - ETA: 4s - loss: 0.4218
678/1875 [=========>....................] - ETA: 4s - loss: 0.4220
693/1875 [==========>...................] - ETA: 4s - loss: 0.4215
708/1875 [==========>...................] - ETA: 4s - loss: 0.4208
723/1875 [==========>...................] - ETA: 4s - loss: 0.4208
737/1875 [==========>...................] - ETA: 3s - loss: 0.4211
751/1875 [===========>..................] - ETA: 3s - loss: 0.4219
766/1875 [===========>..................] - ETA: 3s - loss: 0.4217
781/1875 [===========>..................] - ETA: 3s - loss: 0.4209
796/1875 [===========>..................] - ETA: 3s - loss: 0.4212
811/1875 [===========>..................] - ETA: 3s - loss: 0.4211
826/1875 [============>.................] - ETA: 3s - loss: 0.4215
839/1875 [============>.................] - ETA: 3s - loss: 0.4219
854/1875 [============>.................] - ETA: 3s - loss: 0.4217
869/1875 [============>.................] - ETA: 3s - loss: 0.4211
884/1875 [=============>................] - ETA: 3s - loss: 0.4208
899/1875 [=============>................] - ETA: 3s - loss: 0.4214
914/1875 [=============>................] - ETA: 3s - loss: 0.4212
929/1875 [=============>................] - ETA: 3s - loss: 0.4208
944/1875 [==============>...............] - ETA: 3s - loss: 0.4206
959/1875 [==============>...............] - ETA: 3s - loss: 0.4205
974/1875 [==============>...............] - ETA: 3s - loss: 0.4204
989/1875 [==============>...............] - ETA: 3s - loss: 0.4201
1004/1875 [===============>..............] - ETA: 3s - loss: 0.4204
1020/1875 [===============>..............] - ETA: 2s - loss: 0.4209
1035/1875 [===============>..............] - ETA: 2s - loss: 0.4206
1050/1875 [===============>..............] - ETA: 2s - loss: 0.4200
1065/1875 [================>.............] - ETA: 2s - loss: 0.4197
1081/1875 [================>.............] - ETA: 2s - loss: 0.4201
1096/1875 [================>.............] - ETA: 2s - loss: 0.4201
1111/1875 [================>.............] - ETA: 2s - loss: 0.4207
1126/1875 [=================>............] - ETA: 2s - loss: 0.4204
1141/1875 [=================>............] - ETA: 2s - loss: 0.4204
1155/1875 [=================>............] - ETA: 2s - loss: 0.4202
1170/1875 [=================>............] - ETA: 2s - loss: 0.4207
1184/1875 [=================>............] - ETA: 2s - loss: 0.4214
1199/1875 [==================>...........] - ETA: 2s - loss: 0.4210
1214/1875 [==================>...........] - ETA: 2s - loss: 0.4205
1229/1875 [==================>...........] - ETA: 2s - loss: 0.4203
1244/1875 [==================>...........] - ETA: 2s - loss: 0.4206
1259/1875 [===================>..........] - ETA: 2s - loss: 0.4199
1274/1875 [===================>..........] - ETA: 2s - loss: 0.4201
1289/1875 [===================>..........] - ETA: 2s - loss: 0.4203
1304/1875 [===================>..........] - ETA: 1s - loss: 0.4205
1320/1875 [====================>.........] - ETA: 1s - loss: 0.4197
1335/1875 [====================>.........] - ETA: 1s - loss: 0.4195
1350/1875 [====================>.........] - ETA: 1s - loss: 0.4200
1365/1875 [====================>.........] - ETA: 1s - loss: 0.4197
1381/1875 [=====================>........] - ETA: 1s - loss: 0.4188
1396/1875 [=====================>........] - ETA: 1s - loss: 0.4187
1411/1875 [=====================>........] - ETA: 1s - loss: 0.4189
1426/1875 [=====================>........] - ETA: 1s - loss: 0.4188
1442/1875 [======================>.......] - ETA: 1s - loss: 0.4187
1456/1875 [======================>.......] - ETA: 1s - loss: 0.4191
1470/1875 [======================>.......] - ETA: 1s - loss: 0.4193
1484/1875 [======================>.......] - ETA: 1s - loss: 0.4191
1499/1875 [======================>.......] - ETA: 1s - loss: 0.4192
1513/1875 [=======================>......] - ETA: 1s - loss: 0.4191
1527/1875 [=======================>......] - ETA: 1s - loss: 0.4192
1542/1875 [=======================>......] - ETA: 1s - loss: 0.4192
1556/1875 [=======================>......] - ETA: 1s - loss: 0.4195
1571/1875 [========================>.....] - ETA: 1s - loss: 0.4191
1586/1875 [========================>.....] - ETA: 1s - loss: 0.4193
1602/1875 [========================>.....] - ETA: 0s - loss: 0.4191
1616/1875 [========================>.....] - ETA: 0s - loss: 0.4193
1631/1875 [=========================>....] - ETA: 0s - loss: 0.4193
1646/1875 [=========================>....] - ETA: 0s - loss: 0.4194
1661/1875 [=========================>....] - ETA: 0s - loss: 0.4193
1677/1875 [=========================>....] - ETA: 0s - loss: 0.4191
1692/1875 [==========================>...] - ETA: 0s - loss: 0.4190
1707/1875 [==========================>...] - ETA: 0s - loss: 0.4190
1722/1875 [==========================>...] - ETA: 0s - loss: 0.4187
1737/1875 [==========================>...] - ETA: 0s - loss: 0.4188
1752/1875 [===========================>..] - ETA: 0s - loss: 0.4189
1767/1875 [===========================>..] - ETA: 0s - loss: 0.4187
1782/1875 [===========================>..] - ETA: 0s - loss: 0.4182
1797/1875 [===========================>..] - ETA: 0s - loss: 0.4183
1812/1875 [===========================>..] - ETA: 0s - loss: 0.4180
1827/1875 [============================>.] - ETA: 0s - loss: 0.4175
1842/1875 [============================>.] - ETA: 0s - loss: 0.4175
1856/1875 [============================>.] - ETA: 0s - loss: 0.4175
1871/1875 [============================>.] - ETA: 0s - loss: 0.4174
1875/1875 [==============================] - 7s 3ms/step - loss: 0.4175
Epoch 4/5

   1/1875 [..............................] - ETA: 0s - loss: 0.4156
16/1875 [..............................] - ETA: 5s - loss: 0.4068
31/1875 [..............................] - ETA: 6s - loss: 0.3739
46/1875 [..............................] - ETA: 6s - loss: 0.3828
61/1875 [..............................] - ETA: 6s - loss: 0.3868
76/1875 [>.............................] - ETA: 6s - loss: 0.3960
91/1875 [>.............................] - ETA: 5s - loss: 0.4030
106/1875 [>.............................] - ETA: 5s - loss: 0.4032
120/1875 [>.............................] - ETA: 5s - loss: 0.4014
135/1875 [=>............................] - ETA: 5s - loss: 0.4001
150/1875 [=>............................] - ETA: 5s - loss: 0.3989
165/1875 [=>............................] - ETA: 5s - loss: 0.4009
179/1875 [=>............................] - ETA: 5s - loss: 0.3998
193/1875 [==>...........................] - ETA: 5s - loss: 0.3990
206/1875 [==>...........................] - ETA: 5s - loss: 0.3986
221/1875 [==>...........................] - ETA: 5s - loss: 0.3959
236/1875 [==>...........................] - ETA: 5s - loss: 0.3984
250/1875 [===>..........................] - ETA: 5s - loss: 0.3970
264/1875 [===>..........................] - ETA: 5s - loss: 0.3955
278/1875 [===>..........................] - ETA: 5s - loss: 0.3954
292/1875 [===>..........................] - ETA: 5s - loss: 0.3961
307/1875 [===>..........................] - ETA: 5s - loss: 0.3949
322/1875 [====>.........................] - ETA: 5s - loss: 0.3951
336/1875 [====>.........................] - ETA: 5s - loss: 0.3938
350/1875 [====>.........................] - ETA: 5s - loss: 0.3934
364/1875 [====>.........................] - ETA: 5s - loss: 0.3949
378/1875 [=====>........................] - ETA: 5s - loss: 0.3948
393/1875 [=====>........................] - ETA: 5s - loss: 0.3955
408/1875 [=====>........................] - ETA: 5s - loss: 0.3959
423/1875 [=====>........................] - ETA: 5s - loss: 0.3972
438/1875 [======>.......................] - ETA: 5s - loss: 0.3966
453/1875 [======>.......................] - ETA: 5s - loss: 0.3974
468/1875 [======>.......................] - ETA: 4s - loss: 0.3952
483/1875 [======>.......................] - ETA: 4s - loss: 0.3949
498/1875 [======>.......................] - ETA: 4s - loss: 0.3944
512/1875 [=======>......................] - ETA: 4s - loss: 0.3937
527/1875 [=======>......................] - ETA: 4s - loss: 0.3940
542/1875 [=======>......................] - ETA: 4s - loss: 0.3943
557/1875 [=======>......................] - ETA: 4s - loss: 0.3942
572/1875 [========>.....................] - ETA: 4s - loss: 0.3943
587/1875 [========>.....................] - ETA: 4s - loss: 0.3944
602/1875 [========>.....................] - ETA: 4s - loss: 0.3940
616/1875 [========>.....................] - ETA: 4s - loss: 0.3932
631/1875 [=========>....................] - ETA: 4s - loss: 0.3925
646/1875 [=========>....................] - ETA: 4s - loss: 0.3917
661/1875 [=========>....................] - ETA: 4s - loss: 0.3915
676/1875 [=========>....................] - ETA: 4s - loss: 0.3913
691/1875 [==========>...................] - ETA: 4s - loss: 0.3905
705/1875 [==========>...................] - ETA: 4s - loss: 0.3912
720/1875 [==========>...................] - ETA: 4s - loss: 0.3919
735/1875 [==========>...................] - ETA: 4s - loss: 0.3919
750/1875 [===========>..................] - ETA: 3s - loss: 0.3932
765/1875 [===========>..................] - ETA: 3s - loss: 0.3930
779/1875 [===========>..................] - ETA: 3s - loss: 0.3927
794/1875 [===========>..................] - ETA: 3s - loss: 0.3935
809/1875 [===========>..................] - ETA: 3s - loss: 0.3929
824/1875 [============>.................] - ETA: 3s - loss: 0.3930
839/1875 [============>.................] - ETA: 3s - loss: 0.3927
854/1875 [============>.................] - ETA: 3s - loss: 0.3932
869/1875 [============>.................] - ETA: 3s - loss: 0.3927
884/1875 [=============>................] - ETA: 3s - loss: 0.3927
899/1875 [=============>................] - ETA: 3s - loss: 0.3928
914/1875 [=============>................] - ETA: 3s - loss: 0.3923
927/1875 [=============>................] - ETA: 3s - loss: 0.3924
941/1875 [==============>...............] - ETA: 3s - loss: 0.3927
955/1875 [==============>...............] - ETA: 3s - loss: 0.3933
969/1875 [==============>...............] - ETA: 3s - loss: 0.3938
984/1875 [==============>...............] - ETA: 3s - loss: 0.3942
999/1875 [==============>...............] - ETA: 3s - loss: 0.3946
1014/1875 [===============>..............] - ETA: 3s - loss: 0.3937
1029/1875 [===============>..............] - ETA: 2s - loss: 0.3941
1044/1875 [===============>..............] - ETA: 2s - loss: 0.3938
1059/1875 [===============>..............] - ETA: 2s - loss: 0.3933
1074/1875 [================>.............] - ETA: 2s - loss: 0.3938
1090/1875 [================>.............] - ETA: 2s - loss: 0.3936
1104/1875 [================>.............] - ETA: 2s - loss: 0.3939
1117/1875 [================>.............] - ETA: 2s - loss: 0.3939
1132/1875 [=================>............] - ETA: 2s - loss: 0.3937
1147/1875 [=================>............] - ETA: 2s - loss: 0.3933
1162/1875 [=================>............] - ETA: 2s - loss: 0.3936
1177/1875 [=================>............] - ETA: 2s - loss: 0.3941
1193/1875 [==================>...........] - ETA: 2s - loss: 0.3935
1208/1875 [==================>...........] - ETA: 2s - loss: 0.3934
1224/1875 [==================>...........] - ETA: 2s - loss: 0.3936
1239/1875 [==================>...........] - ETA: 2s - loss: 0.3939
1254/1875 [===================>..........] - ETA: 2s - loss: 0.3933
1269/1875 [===================>..........] - ETA: 2s - loss: 0.3937
1284/1875 [===================>..........] - ETA: 2s - loss: 0.3935
1299/1875 [===================>..........] - ETA: 2s - loss: 0.3937
1315/1875 [====================>.........] - ETA: 1s - loss: 0.3939
1330/1875 [====================>.........] - ETA: 1s - loss: 0.3940
1346/1875 [====================>.........] - ETA: 1s - loss: 0.3938
1359/1875 [====================>.........] - ETA: 1s - loss: 0.3934
1374/1875 [====================>.........] - ETA: 1s - loss: 0.3930
1389/1875 [=====================>........] - ETA: 1s - loss: 0.3933
1404/1875 [=====================>........] - ETA: 1s - loss: 0.3940
1419/1875 [=====================>........] - ETA: 1s - loss: 0.3945
1435/1875 [=====================>........] - ETA: 1s - loss: 0.3947
1450/1875 [======================>.......] - ETA: 1s - loss: 0.3941
1466/1875 [======================>.......] - ETA: 1s - loss: 0.3938
1481/1875 [======================>.......] - ETA: 1s - loss: 0.3941
1495/1875 [======================>.......] - ETA: 1s - loss: 0.3941
1510/1875 [=======================>......] - ETA: 1s - loss: 0.3940
1525/1875 [=======================>......] - ETA: 1s - loss: 0.3939
1540/1875 [=======================>......] - ETA: 1s - loss: 0.3936
1554/1875 [=======================>......] - ETA: 1s - loss: 0.3937
1569/1875 [========================>.....] - ETA: 1s - loss: 0.3936
1584/1875 [========================>.....] - ETA: 1s - loss: 0.3941
1599/1875 [========================>.....] - ETA: 0s - loss: 0.3937
1615/1875 [========================>.....] - ETA: 0s - loss: 0.3938
1631/1875 [=========================>....] - ETA: 0s - loss: 0.3944
1646/1875 [=========================>....] - ETA: 0s - loss: 0.3945
1661/1875 [=========================>....] - ETA: 0s - loss: 0.3942
1676/1875 [=========================>....] - ETA: 0s - loss: 0.3943
1691/1875 [==========================>...] - ETA: 0s - loss: 0.3945
1706/1875 [==========================>...] - ETA: 0s - loss: 0.3944
1720/1875 [==========================>...] - ETA: 0s - loss: 0.3941
1735/1875 [==========================>...] - ETA: 0s - loss: 0.3940
1749/1875 [==========================>...] - ETA: 0s - loss: 0.3937
1764/1875 [===========================>..] - ETA: 0s - loss: 0.3934
1779/1875 [===========================>..] - ETA: 0s - loss: 0.3931
1793/1875 [===========================>..] - ETA: 0s - loss: 0.3934
1807/1875 [===========================>..] - ETA: 0s - loss: 0.3936
1822/1875 [============================>.] - ETA: 0s - loss: 0.3936
1838/1875 [============================>.] - ETA: 0s - loss: 0.3941
1853/1875 [============================>.] - ETA: 0s - loss: 0.3938
1868/1875 [============================>.] - ETA: 0s - loss: 0.3939
1875/1875 [==============================] - 7s 3ms/step - loss: 0.3938
Epoch 5/5
1/1875 [..............................] - ETA: 0s - loss: 0.5834
16/1875 [..............................] - ETA: 5s - loss: 0.4310
30/1875 [..............................] - ETA: 6s - loss: 0.4008
45/1875 [..............................] - ETA: 6s - loss: 0.4012
60/1875 [..............................] - ETA: 6s - loss: 0.4037
73/1875 [>.............................] - ETA: 6s - loss: 0.4036
87/1875 [>.............................] - ETA: 6s - loss: 0.4039
101/1875 [>.............................] - ETA: 6s - loss: 0.4062
116/1875 [>.............................] - ETA: 6s - loss: 0.3992
130/1875 [=>............................] - ETA: 6s - loss: 0.4046
145/1875 [=>............................] - ETA: 6s - loss: 0.4002
160/1875 [=>............................] - ETA: 6s - loss: 0.3994
174/1875 [=>............................] - ETA: 6s - loss: 0.3921
189/1875 [==>...........................] - ETA: 5s - loss: 0.3902
204/1875 [==>...........................] - ETA: 5s - loss: 0.3944
219/1875 [==>...........................] - ETA: 5s - loss: 0.3936
234/1875 [==>...........................] - ETA: 5s - loss: 0.3924
249/1875 [==>...........................] - ETA: 5s - loss: 0.3940
264/1875 [===>..........................] - ETA: 5s - loss: 0.3923
279/1875 [===>..........................] - ETA: 5s - loss: 0.3907
293/1875 [===>..........................] - ETA: 5s - loss: 0.3926
308/1875 [===>..........................] - ETA: 5s - loss: 0.3917
323/1875 [====>.........................] - ETA: 5s - loss: 0.3906
338/1875 [====>.........................] - ETA: 5s - loss: 0.3894
353/1875 [====>.........................] - ETA: 5s - loss: 0.3902
368/1875 [====>.........................] - ETA: 5s - loss: 0.3891
383/1875 [=====>........................] - ETA: 5s - loss: 0.3885
398/1875 [=====>........................] - ETA: 5s - loss: 0.3910
413/1875 [=====>........................] - ETA: 5s - loss: 0.3936
428/1875 [=====>........................] - ETA: 5s - loss: 0.3931
443/1875 [======>.......................] - ETA: 4s - loss: 0.3929
459/1875 [======>.......................] - ETA: 4s - loss: 0.3929
473/1875 [======>.......................] - ETA: 4s - loss: 0.3921
488/1875 [======>.......................] - ETA: 4s - loss: 0.3909
502/1875 [=======>......................] - ETA: 4s - loss: 0.3905
516/1875 [=======>......................] - ETA: 4s - loss: 0.3915
531/1875 [=======>......................] - ETA: 4s - loss: 0.3918
546/1875 [=======>......................] - ETA: 4s - loss: 0.3904
562/1875 [=======>......................] - ETA: 4s - loss: 0.3918
577/1875 [========>.....................] - ETA: 4s - loss: 0.3919
591/1875 [========>.....................] - ETA: 4s - loss: 0.3914
606/1875 [========>.....................] - ETA: 4s - loss: 0.3904
621/1875 [========>.....................] - ETA: 4s - loss: 0.3894
636/1875 [=========>....................] - ETA: 4s - loss: 0.3896
651/1875 [=========>....................] - ETA: 4s - loss: 0.3893
665/1875 [=========>....................] - ETA: 4s - loss: 0.3879
679/1875 [=========>....................] - ETA: 4s - loss: 0.3881
694/1875 [==========>...................] - ETA: 4s - loss: 0.3874
709/1875 [==========>...................] - ETA: 4s - loss: 0.3873
724/1875 [==========>...................] - ETA: 4s - loss: 0.3870
738/1875 [==========>...................] - ETA: 3s - loss: 0.3870
752/1875 [===========>..................] - ETA: 3s - loss: 0.3873
767/1875 [===========>..................] - ETA: 3s - loss: 0.3872
783/1875 [===========>..................] - ETA: 3s - loss: 0.3877
798/1875 [===========>..................] - ETA: 3s - loss: 0.3875
813/1875 [============>.................] - ETA: 3s - loss: 0.3880
828/1875 [============>.................] - ETA: 3s - loss: 0.3888
842/1875 [============>.................] - ETA: 3s - loss: 0.3889
856/1875 [============>.................] - ETA: 3s - loss: 0.3887
869/1875 [============>.................] - ETA: 3s - loss: 0.3889
884/1875 [=============>................] - ETA: 3s - loss: 0.3901
899/1875 [=============>................] - ETA: 3s - loss: 0.3900
914/1875 [=============>................] - ETA: 3s - loss: 0.3905
928/1875 [=============>................] - ETA: 3s - loss: 0.3895
944/1875 [==============>...............] - ETA: 3s - loss: 0.3900
959/1875 [==============>...............] - ETA: 3s - loss: 0.3904
974/1875 [==============>...............] - ETA: 3s - loss: 0.3911
989/1875 [==============>...............] - ETA: 3s - loss: 0.3904
1004/1875 [===============>..............] - ETA: 3s - loss: 0.3906
1019/1875 [===============>..............] - ETA: 2s - loss: 0.3910
1034/1875 [===============>..............] - ETA: 2s - loss: 0.3908
1049/1875 [===============>..............] - ETA: 2s - loss: 0.3910
1064/1875 [================>.............] - ETA: 2s - loss: 0.3912
1079/1875 [================>.............] - ETA: 2s - loss: 0.3913
1094/1875 [================>.............] - ETA: 2s - loss: 0.3915
1108/1875 [================>.............] - ETA: 2s - loss: 0.3919
1123/1875 [================>.............] - ETA: 2s - loss: 0.3919
1138/1875 [=================>............] - ETA: 2s - loss: 0.3922
1153/1875 [=================>............] - ETA: 2s - loss: 0.3924
1168/1875 [=================>............] - ETA: 2s - loss: 0.3921
1182/1875 [=================>............] - ETA: 2s - loss: 0.3919
1197/1875 [==================>...........] - ETA: 2s - loss: 0.3916
1212/1875 [==================>...........] - ETA: 2s - loss: 0.3921
1226/1875 [==================>...........] - ETA: 2s - loss: 0.3919
1239/1875 [==================>...........] - ETA: 2s - loss: 0.3921
1254/1875 [===================>..........] - ETA: 2s - loss: 0.3918
1269/1875 [===================>..........] - ETA: 2s - loss: 0.3915
1284/1875 [===================>..........] - ETA: 2s - loss: 0.3914
1299/1875 [===================>..........] - ETA: 2s - loss: 0.3912
1314/1875 [====================>.........] - ETA: 1s - loss: 0.3913
1329/1875 [====================>.........] - ETA: 1s - loss: 0.3916
1344/1875 [====================>.........] - ETA: 1s - loss: 0.3917
1359/1875 [====================>.........] - ETA: 1s - loss: 0.3911
1374/1875 [====================>.........] - ETA: 1s - loss: 0.3910
1388/1875 [=====================>........] - ETA: 1s - loss: 0.3909
1404/1875 [=====================>........] - ETA: 1s - loss: 0.3911
1419/1875 [=====================>........] - ETA: 1s - loss: 0.3911
1434/1875 [=====================>........] - ETA: 1s - loss: 0.3912
1449/1875 [======================>.......] - ETA: 1s - loss: 0.3912
1465/1875 [======================>.......] - ETA: 1s - loss: 0.3914
1480/1875 [======================>.......] - ETA: 1s - loss: 0.3918
1496/1875 [======================>.......] - ETA: 1s - loss: 0.3915
1511/1875 [=======================>......] - ETA: 1s - loss: 0.3914
1527/1875 [=======================>......] - ETA: 1s - loss: 0.3916
1542/1875 [=======================>......] - ETA: 1s - loss: 0.3919
1557/1875 [=======================>......] - ETA: 1s - loss: 0.3917
1572/1875 [========================>.....] - ETA: 1s - loss: 0.3920
1588/1875 [========================>.....] - ETA: 0s - loss: 0.3922
1603/1875 [========================>.....] - ETA: 0s - loss: 0.3918
1618/1875 [========================>.....] - ETA: 0s - loss: 0.3919
1633/1875 [=========================>....] - ETA: 0s - loss: 0.3916
1648/1875 [=========================>....] - ETA: 0s - loss: 0.3918
1663/1875 [=========================>....] - ETA: 0s - loss: 0.3917
1678/1875 [=========================>....] - ETA: 0s - loss: 0.3918
1693/1875 [==========================>...] - ETA: 0s - loss: 0.3918
1708/1875 [==========================>...] - ETA: 0s - loss: 0.3913
1723/1875 [==========================>...] - ETA: 0s - loss: 0.3915
1737/1875 [==========================>...] - ETA: 0s - loss: 0.3913
1752/1875 [===========================>..] - ETA: 0s - loss: 0.3909
1767/1875 [===========================>..] - ETA: 0s - loss: 0.3906
1782/1875 [===========================>..] - ETA: 0s - loss: 0.3906
1797/1875 [===========================>..] - ETA: 0s - loss: 0.3902
1812/1875 [===========================>..] - ETA: 0s - loss: 0.3899
1827/1875 [============================>.] - ETA: 0s - loss: 0.3898
1842/1875 [============================>.] - ETA: 0s - loss: 0.3901
1856/1875 [============================>.] - ETA: 0s - loss: 0.3898
1871/1875 [============================>.] - ETA: 0s - loss: 0.3899
1875/1875 [==============================] - 7s 3ms/step - loss: 0.3898
# Evaluate the network
results = model.predict(test_dataset)
# Save test embeddings for visualization in projector
np.savetxt("vecs.tsv", results, delimiter='\t')

out_m = io.open('meta.tsv', 'w', encoding='utf-8')
for img, labels in tfds.as_numpy(test_dataset):
    [out_m.write(str(x) + "\n") for x in labels]
out_m.close()


try:
  from google.colab import files
  files.download('vecs.tsv')
  files.download('meta.tsv')
except:
  pass

Embedding Projector

벡터 및 메타 데이터 파일은 https://projector.tensorflow.org/에서 로드하고 시각화할 수 있습니다.

UMAP으로 시각화하면 포함된 테스트 데이터의 결과를 볼 수 있습니다. embedding