No project description provided
Project description
Torch Lure
Depndencies
pip install git+https://github.com/Farama-Foundation/Minari.git@19565bd8cd33f2e4a3a9a8e4db372044b01ea8d3
pip install torchlure
Usage
import torchlure as lure
# Optimizers
lure.SophiaG(lr=1e-3, weight_decay=0.2)
# Functions
lure.tanh_exp(x)
lure.TanhExp()
lure.quantile_loss(y_pred, y_target, quantile=0.5)
lure.QuantileLoss(quantile=0.5)
lure.RMSNrom(dim=256, eps=1e-6)
# Noise Scheduler
lure.LinearNoiseScheduler(beta=1e-4, beta_end=0.02, num_timesteps=1000)
lure.CosineNoiseScheduler(max_beta=0.999, s=0.008, num_timesteps=1000):
Dataset
from torchlure.datasets import OfflineRLDataset, D4RLDataset
# Initial usage with download
dataset = D4RLDataset(
dataset_id="d4rl_walker2d-expert-2405",
d4rl_name="d4rl_walker2d-expert-v2",
env_id= "Walker2d-v4",
)
# if you are download it once
dataset = D4RLDataset(
dataset_id="d4rl_walker2d-expert-2405",
)
See all datasets here
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distributions
No source distribution files available for this release.See tutorial on generating distribution archives.
Built Distribution
Close
Hashes for torchlure-0.2405.6-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 9da042d9e6f82835a762c20969b25f9b70a8049365b6b5a7e58794865b1e7810 |
|
MD5 | d23aaf3afbe8821c3fc1a64d5e445886 |
|
BLAKE2b-256 | 11818417add22128bc218ad85be5ee57d02a60423ae0ae63e8c78934756d7208 |