Skip to content

Commit f863411

Browse files
MarkDaoustcopybara-github
authored andcommitted
Update image captioning
- Switch to a transformer architecture. - Use a smaller, more-manageable dataset. - Use `tf.data.Dataset.save` to cache features. - Add a nice TokenOutput layer. - Print predictions during training. - Train with `Model.fit` - Clear all the unnecessary mask handling. PiperOrigin-RevId: 479585366
1 parent 057d9b0 commit f863411

File tree

1 file changed

+1676
-557
lines changed

1 file changed

+1676
-557
lines changed

0 commit comments

Comments
 (0)