Closed tiancheng2000 closed 4 years ago
Hi, original author and reviewers, I re-run my TL2 version recently, and found that even after a 6-hour model training, the weights was already good enough to produce some impressive outputs (as show below). HTH!
@tiancheng2000 you can merge the PR, if you are sure others can run the code.
Migration points:
@tf.function
) train step. (key points for TL2 migration)tensorlayer\models\vgg19.py
instead of handmade.save_weights/load_weights
instead oftl.files.save_npz/load_npz
.tf.data.Database