hongshaorou/Tianshou
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix WandbLogger import error in Atari examples (#562)

Browse Source
This commit is contained in:
Chengqi Duan 2022-03-08 08:38:56 -05:00
parent df3d7f582b
commit ad2e1eaea0
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
examples/atari/atari_qrdqn.py
View File

@ -7,12 +7,12 @@ import numpy as np
import torch import torch
from atari_network import QRDQN from atari_network import QRDQN
from atari_wrapper import make_atari_env from atari_wrapper import make_atari_env
from torch.utils.tensorboard import SummaryWriter, WandbLogger from torch.utils.tensorboard import SummaryWriter
from tianshou.data import Collector, VectorReplayBuffer from tianshou.data import Collector, VectorReplayBuffer
from tianshou.policy import QRDQNPolicy from tianshou.policy import QRDQNPolicy
from tianshou.trainer import offpolicy_trainer from tianshou.trainer import offpolicy_trainer
from tianshou.utils import TensorboardLogger from tianshou.utils import TensorboardLogger, WandbLogger
def get_args(): def get_args():

4
examples/atari/atari_rainbow.py
View File

@ -7,12 +7,12 @@ import numpy as np
import torch import torch
from atari_network import Rainbow from atari_network import Rainbow
from atari_wrapper import make_atari_env from atari_wrapper import make_atari_env
from torch.utils.tensorboard import SummaryWriter, WandbLogger from torch.utils.tensorboard import SummaryWriter
from tianshou.data import Collector, PrioritizedVectorReplayBuffer, VectorReplayBuffer from tianshou.data import Collector, PrioritizedVectorReplayBuffer, VectorReplayBuffer
from tianshou.policy import RainbowPolicy from tianshou.policy import RainbowPolicy
from tianshou.trainer import offpolicy_trainer from tianshou.trainer import offpolicy_trainer
from tianshou.utils import TensorboardLogger from tianshou.utils import TensorboardLogger, WandbLogger
def get_args(): def get_args():