dragen1860 / TensorFlow-2.x-Tutorials

TensorFlow 2.x version's Tutorials and Examples, including CNN, RNN, GAN, Auto-Encoders, FasterRCNN, GPT, BERT examples, etc. TF 2.0版入门实例代码,实战教程。
6.38k stars 2.24k forks source link

Update main.py #58

Open wenqinlong opened 3 years ago

wenqinlong commented 3 years ago

The shape of y is (batch_size, 10), so you don't need to encode it any more.