訓練可能な変数は2つ : 全結合層の weights と bias
>>> len(model.trainable_variables)
2
モデルの訓練は,ここではデモが目的ということで,steps_per_epoch=3 とする:
>>>model.fit(ds, epochs=1, steps_per_epoch=3)
3/3 [==============================] - 9s 39ms/step - loss: 6.5376 - accuracy: 0.2630
註. 普通は,steps_per_epoch をつぎのように定める:
>>> steps_per_epoch = \
... tf.math.ceil(len(all_image_paths)/BATCH_SIZE).numpy()
>>> steps_per_epoch
115.0
|