search

ClownW / Reinforcement-learning-with-PyTorch

Reinforcement learning with PyTorch, inspired by MorvanZhou, change the framework from Tensorflow to PyTorch
229 stars 64 forks source link

Using tensorboard to show graph of DQN #2

Open jzndd opened 1 year ago

jzndd commented 1 year ago

在莫烦的网课里使用了 tensorboard来展示网络结构，想请教一下pytorch如何使用tensorboard来展示网络结构（看过很多教程，都是在add_graph方法中传入module，但DQN类并不是一个module，而是由两个module组成的，所以想请教一下有无解决方案）

ClownW commented 1 year ago

您的来信已收到，祝您生活愉快！