maxtext
maxtext copied to clipboard
a little error
Thanks for your contribution, I found a small error in the code: https://github.com/AI-Hypercomputer/maxtext/blob/121293bd3fcdff6cb676a7331c21f668cbdc3482/MaxText/train.py#L267 metrics_dict = intermediate_outputs["intermediates"]["decoder"]["decoder"] should be: metrics_dict = intermediate_outputs["intermediates"]["decoder"]["layers"] ?
@Lisennlp thank you for reporting this issue! we were curious whether you use this code ? can you provide some info about your use-case