연구와 실제 응용에 신뢰할 수 있고 일관된 기준선을 제공하도록 설계된 강화 학습 알고리즘의 PyTorch 구현체입니다.
DLR-RM/rl-baselines3-zoo
튜닝된 하이퍼파라미터와 사전 학습 에이전트, 하이퍼파라미터 최적화 스크립트를 제공하는 Stable Baselines3 강화 학습 에이전트용 학습 프레임워크입니다.
- 스타
- 2,900
- 포크
- 612
- 열린 이슈
- 62
- 마지막 커밋
- 5일 전
어떤 프로젝트인가요?
RL Baselines3 Zoo는 Stable Baselines3 기반의 강화 학습용 학습 프레임워크입니다. 에이전트 학습 및 평가, 하이퍼파라미터 튜닝, 결과 시각화, 에이전트 동영상 녹화를 위한 CLI 스크립트를 제공합니다.
또한 일반적인 환경과 알고리즘에 맞춰 튜닝된 하이퍼파라미터 모음과 해당 설정으로 학습한 에이전트를 제공합니다. 하이퍼파라미터는 환경별 YAML 파일에 정의되어 있어, 알고리즘과 환경 ID를 지정하는 등의 단일 명령으로 학습을 시작할 수 있습니다. 학습된 에이전트는 enjoy 스크립트로 재생할 수 있으며, 저장소는 Atari, classic control, Box2D, PyBullet, MuJoCo, robotics, MiniGrid 환경을 다룹니다.
실험 추적을 위한 Weights & Biases 및 학습된 모델의 저장과 공유를 위한 Hugging Face 연동을 지원합니다. RL 에이전트를 간단히 학습하고, 알고리즘을 벤치마크하며, 사전 학습된 설정을 재사용하려는 사용자를 위한 프로젝트입니다.
핵심 기능
- 01에이전트 학습, 평가, 재생을 위한 CLI 스크립트
- 02환경과 알고리즘별로 YAML 파일에 정의된 튜닝된 하이퍼파라미터
- 03하이퍼파라미터 최적화 도구
- 04Hugging Face에서 사용할 수 있는 사전 학습 에이전트
- 05Weights & Biases 및 Hugging Face 연동
- 06학습 결과 시각화 스크립트
활용 사례
- 보일러플레이트 코드를 작성하지 않고 Gym 또는 PyBullet 환경에서 RL 에이전트 학습
- 같은 작업에서 여러 RL 알고리즘 벤치마크
- 새로운 환경에 적합한 하이퍼파라미터 탐색
- 사전 학습 에이전트를 재사용해 정책을 확인하거나 평가
빠른 시작
- 1
PyPI에서 rl_zoo3 패키지를 설치합니다
pip install rl_zoo3 - 2
복제한 저장소에서 주기적으로 평가를 진행하며 HalfCheetahBulletEnv-v0 환경의 SAC 에이전트를 학습시킵니다
python train.py --algo sac --env HalfCheetahBulletEnv-v0 --eval-freq 10000 --eval-episodes 10 --n-eval-envs 1
사용 전 알아둘 점
- 전체 설치에는 swig, cmake, ffmpeg 등의 시스템 패키지가 필요합니다.
- README에서는 특별한 CFLAGS 우회 방법을 사용하지 않으면 macOS에서 pybullet 빌드가 실패할 수 있다고 경고합니다.
- 벤치마크 결과는 각각 한 번의 실행에서 나온 것이므로 정량적 벤치마크가 아닙니다.
토픽
자주 묻는 질문
- rl-baselines3-zoo은(는) 무엇인가요?
- 튜닝된 하이퍼파라미터와 사전 학습 에이전트, 하이퍼파라미터 최적화 스크립트를 제공하는 Stable Baselines3 강화 학습 에이전트용 학습 프레임워크입니다. RL Baselines3 Zoo는 Stable Baselines3 기반의 강화 학습용 학습 프레임워크입니다. 에이전트 학습 및 평가, 하이퍼파라미터 튜닝, 결과 시각화, 에이전트 동영상 녹화를 위한 CLI 스크립트를 제공합니다.
- rl-baselines3-zoo은(는) 무료로 사용할 수 있나요? 라이선스는 무엇인가요?
- rl-baselines3-zoo은(는) MIT 라이선스의 오픈소스입니다. 상업용 제품에 사용하기 전에 라이선스 조건을 확인하세요.
- rl-baselines3-zoo은(는) 아직 유지보수되고 있나요?
- 활발히 유지보수 중. rl-baselines3-zoo의 마지막 커밋은 2026년 10월 16일입니다.
- rl-baselines3-zoo은(는) 어디에 사용할 수 있나요?
- 보일러플레이트 코드를 작성하지 않고 Gym 또는 PyBullet 환경에서 RL 에이전트 학습 같은 작업에서 여러 RL 알고리즘 벤치마크 새로운 환경에 적합한 하이퍼파라미터 탐색 사전 학습 에이전트를 재사용해 정책을 확인하거나 평가
- rl-baselines3-zoo는 어떻게 설치하나요?
- PyPI에서 rl_zoo3 패키지를 설치합니다: pip install rl_zoo3 복제한 저장소에서 주기적으로 평가를 진행하며 HalfCheetahBulletEnv-v0 환경의 SAC 에이전트를 학습시킵니다: python train.py --algo sac --env HalfCheetahBulletEnv-v0 --eval-freq 10000 --eval-episodes 10 --n-eval-envs 1
- rl-baselines3-zoo은(는) 어디서 받을 수 있나요?
- 소스 코드와 설치 방법은 GitHub에 있습니다: https://github.com/DLR-RM/rl-baselines3-zoo
요약은 프로젝트 README를 바탕으로 AI가 작성했습니다. GitHub 통계는 매일 갱신됩니다.
관련 프로젝트
자동 주식 거래와 포트폴리오 작업에 딥 강화학습을 적용하기 위한 오픈소스 Python 프레임워크입니다.
Physical Intelligence의 오픈소스 로보틱스 VLA 모델로, 사전 학습 체크포인트와 파인튜닝 예제를 제공합니다.
단일 에이전트 강화 학습 환경을 위한 표준 Python API와 참조 환경 모음으로, OpenAI Gym에서 포크되었습니다.
Mamba는 선택적 상태 공간 모델을 기반으로 한 PyTorch 시퀀스 모델링 아키텍처로, 언어를 비롯한 시퀀스 작업에서 Transformer를 대체할 선형 시간 연산 방식을 제공합니다.
단일 24GB GPU에서 실행할 수 있는 소형 중국어 LLaMA2 스타일 모델의 사전학습 및 SFT 파이프라인을 처음부터 제공합니다.