Trax 在我的模型中出现了错误 TypeError: 在 gather op 中的切片大小在索引 0 处超出范围,必须在 [0, 1) 内,但实际为 1。

!rm -f ~/model/model.pkl.gz loop = training_loop(TransformerLM, train_batch_stream, eval_batch_stream) loop.run(10000) 其中 training_loop 的代码定义如下：

training_loop(TransformerLM, train_gen, eval_gen, output_dir = "~/model") 而 "TransformerLM" 是一个模型。 因此，我获得了 0.233 的评估准确率。 为了评估模型的性能，我将模型模式更改为 "eval"，并继续测试我的模型。 再次，评估模型是课程的一部分，教师准备了一个代码块，让我加载提供的模型（该模型应该由教师训练，并且应该比我的模型要好得多）。 我阻止了加载部分，并按照以下方式继续评估部分。