search

rail-berkeley / serl

SERL: A Software Suite for Sample-Efficient Robotic Reinforcement Learning
https://serl-robot.github.io/
MIT License
381 stars 44 forks source link

Img augmentation for drq and bc agent #67

Closed youliangtan closed 4 days ago

youliangtan commented 4 months ago

As described:

image augmentation for drq or bc agent is enabled via

agent.create(
        image_augmentation=("random_crop", "color_transform"),
)