Jiayi Weng
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?
I can push gogoduan.
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?