We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 94e3550 commit 4f7a02cCopy full SHA for 4f7a02c
examples/discrete/discrete_dqn.py
@@ -1,12 +1,13 @@
1
import gymnasium as gym
2
+from torch.utils.tensorboard import SummaryWriter
3
+
4
import tianshou as ts
5
from tianshou.algorithm.modelfree.dqn import DiscreteQLearningPolicy
6
from tianshou.algorithm.optim import AdamOptimizerFactory
7
from tianshou.data import CollectStats
8
from tianshou.trainer import OffPolicyTrainerParams
9
from tianshou.utils.net.common import Net
10
from tianshou.utils.space_info import SpaceInfo
-from torch.utils.tensorboard import SummaryWriter
11
12
13
def main() -> None:
0 commit comments