Gato-A-Generalist-Agent

Gato-A-Generalist-Agent copied to clipboard

你好非常感谢你共享出如此清晰，可读性那么高的代码供我学习。但是在代码中有一处我没有想太明白，想请教一下。 在model.py文件DecisionTransformer类的forward函数中最后对动作、状态进行预测哪一部分。 ``` h = self.embed_ln(h) # transformer and prediction h = self.transformer(h) # 修改 h 的形状, 使得 [B, 3 * T, H] -> [B, 3, T, H] # h[:,...

le-wei