When using
model.compile(optimizer = tf.train.AdamOptimizer(), loss = \'sparse_categorical_crossentropy\', metrics=[\'accuracy\'
It's a minor change in upgraded version.
Please use:
model.compile(optimizer=tf.optimizers.Adam(), loss="sparse_categorical_crossentropy")
Thanks!