avg_checkpoints with tf2.0 ?

tensorflow / tensor2tensor

Library of deep learning models and datasets designed to make deep learning more accessible and accelerate ML research.

Apache License 2.0

15.5k stars 3.49k forks source link

avg_checkpoints with tf2.0 ? #1775

Open leixiaoning opened 4 years ago

leixiaoning commented 4 years ago

tensor2tensor/tensor2tensor/utils/avg_checkpoints.py

i got a trouble in using average ckpt with tf2.0: a variable named 'xx/xx/.xx/xx' , how can it be save in ckpt? i tried, the name is always not correct !

leixiaoning commented 4 years ago

tensor2tensor/tensor2tensor/utils/avg_checkpoints.py

i got a trouble in using average ckpt with tf2.0: a variable named 'xx/xx/.xx/xx' , how can it be save in ckpt? i tried, the name is always not correct !

with tf.train.Checkpoint

yjiangling commented 1 year ago

same question, still no answer?

yjiangling commented 1 year ago

tensor2tensor/tensor2tensor/utils/avg_checkpoints.py i got a trouble in using average ckpt with tf2.0: a variable named 'xx/xx/.xx/xx' , how can it be save in ckpt? i tried, the name is always not correct !

with tf.train.Checkpoint

use tf.train.Checkpoint? How to use it? Any example? Thanks a lot in advance!!!