forked from pytorch/rl
-
Notifications
You must be signed in to change notification settings - Fork 0
/
__init__.py
63 lines (62 loc) · 1.41 KB
/
__init__.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
# Copyright (c) Meta Platforms, Inc. and affiliates.
#
# This source code is licensed under the MIT license found in the
# LICENSE file in the root directory of this source tree.
from .gym_transforms import EndOfLifeTransform
from .r3m import R3MTransform
from .rb_transforms import MultiStepTransform
from .rlhf import KLRewardTransform
from .transforms import (
ActionDiscretizer,
ActionMask,
AutoResetEnv,
AutoResetTransform,
BatchSizeTransform,
BinarizeReward,
BurnInTransform,
CatFrames,
CatTensors,
CenterCrop,
ClipTransform,
Compose,
Crop,
DeviceCastTransform,
DiscreteActionProjection,
DoubleToFloat,
DTypeCastTransform,
ExcludeTransform,
FiniteTensorDictCheck,
FlattenObservation,
FrameSkipTransform,
GrayScale,
gSDENoise,
InitTracker,
NoopResetEnv,
ObservationNorm,
ObservationTransform,
PermuteTransform,
PinMemoryTransform,
RandomCropTensorDict,
RemoveEmptySpecs,
RenameTransform,
Resize,
Reward2GoTransform,
RewardClipping,
RewardScaling,
RewardSum,
SelectTransform,
SignTransform,
SqueezeTransform,
StepCounter,
TargetReturn,
TensorDictPrimer,
TimeMaxPool,
ToTensorImage,
Transform,
TransformedEnv,
UnsqueezeTransform,
VecGymEnvTransform,
VecNorm,
)
from .vc1 import VC1Transform
from .vip import VIPRewardTransform, VIPTransform