Module.tabulate to return the Table

cgarciae / treex

A Pytree Module system for Deep Learning in JAX

https://cgarciae.github.io/treex/

MIT License

215 stars 17 forks source link

Open soumik12345 opened 2 years ago

soumik12345 commented 2 years ago

Updated treex.Module.tabulate to accept an additional parameter return_table which when set to True would return the rich.table.Table object.

cgarciae commented 2 years ago

Thanks @soumik12345!

cgarciae commented 2 years ago

@soumik12345 left a comment.