Jiayi Weng

Results 303 comments of Jiayi Weng

It seems like all tensors in `batch.obs` are on `cuda:0` device. Did you customize the network part? Could you please paste a minimal working example to illustrate this issue? Thanks!

Close due to no response, feel free to re-open this issue.

> Decision Transformer is a recurrent/autoregressive-style Should we first fix tons of issues for RNN?

gogoduan says he will be available after 5.12 and through all summer break.

> I would be interested in this. Thanks in advance!

> I believe something is indeed wrong #486

Feel free to inherit C51Policy: ```python class C51Visualize(C51Policy): def __init__(self, summary_writer, *args, **kwargs): super().__init__(*args, **kwargs) # native tensorboard SummaryWriter, not tianshou.utils.logger.TensorboardLogger # just to demonstrate how to customize logging process...

My bad. ```diff - def forward(self, batch): - result = super().forward(batch) + def forward(self, batch, **kwargs): + result = super().forward(batch, **kwargs) ```

Could you please paste your code here to reproduce the error?