Reinforcement-learning-with-PyTorch icon indicating copy to clipboard operation
Reinforcement-learning-with-PyTorch copied to clipboard

Using tensorboard to show graph of DQN

Open jzndd opened this issue 3 years ago • 1 comments

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

jzndd avatar Aug 21 '22 12:08 jzndd

您的来信已收到,祝您生活愉快!

ClownW avatar Aug 21 '22 12:08 ClownW