Add HourglassLM - Githubissues

google / trax

Trax — Deep Learning with Clear Code and Speed

Apache License 2.0

8.07k stars 813 forks source link

Closed syzymon closed 2 years ago

syzymon commented 3 years ago

HourglassLM implementation, gin configs cleanup + two bugfixes

Duplicate inputs in generate_sequential_chunks when examples are shorter than chunk length.
PrintShape didn't work for n_in = 1 - it split the tensor along the batch axis, printing wrong shapes.