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
from atari_network import QRDQN
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.policy import QRDQNPolicy
from tianshou.trainer import offpolicy_trainer
from tianshou.utils import TensorboardLogger
from tianshou.utils import TensorboardLogger, WandbLogger
def get_args():

4
examples/atari/atari_rainbow.py
View File

@ -7,12 +7,12 @@ import numpy as np
import torch
from atari_network import Rainbow
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.policy import RainbowPolicy
from tianshou.trainer import offpolicy_trainer
from tianshou.utils import TensorboardLogger
from tianshou.utils import TensorboardLogger, WandbLogger
def get_args():