File tree Expand file tree Collapse file tree 1 file changed +5
-3
lines changed Expand file tree Collapse file tree 1 file changed +5
-3
lines changed Original file line number Diff line number Diff line change 39
39
# - how to evaluate your model.
40
40
#
41
41
# Prerequisites
42
- # ~~~~~~~~~~
42
+ # ~~~~~~~~~~~~~
43
43
#
44
44
# This tutorial assumes that you have completed the
45
45
# `PPO tutorial <reinforcement_ppo.html>`_ which gives
59
59
# -----------------
60
60
#
61
61
62
+ import torchrl
63
+
62
64
# sphinx_gallery_start_ignore
63
65
import warnings
64
66
from typing import Tuple
69
71
import torch .cuda
70
72
import tqdm
71
73
72
- import torchrl
73
-
74
74
import torch .multiprocessing
75
75
76
76
###############################################################################
@@ -821,7 +821,9 @@ def make_ddpg_actor(
821
821
# - the total number of frames before the collector is considered empty,
822
822
# - the maximum number of frames per trajectory (useful for non-terminating
823
823
# environments, like ``dm_control`` ones).
824
+ #
824
825
# .. note::
826
+ #
825
827
# The ``max_frames_per_traj`` passed to the collector will have the effect
826
828
# of registering a new :class:`~torchrl.envs.StepCounter` transform
827
829
# with the environment used for inference. We can achieve the same result
You can’t perform that action at this time.
0 commit comments