No project description provided
Project description
Torch Lure
Depndencies
pip install git+https://github.com/Farama-Foundation/Minari.git@19565bd8cd33f2e4a3a9a8e4db372044b01ea8d3
pip install torch-lure
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):
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 Distribution
torchlure-0.2405.2.tar.gz
(10.3 kB
view details)
Built Distribution
File details
Details for the file torchlure-0.2405.2.tar.gz
.
File metadata
- Download URL: torchlure-0.2405.2.tar.gz
- Upload date:
- Size: 10.3 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.2 CPython/3.12.2
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | 2be614c78e26365baa1e4bc23d0655e999607ae946242e4aab73b7754d796612 |
|
MD5 | 110184790158ca6e7c4624d2a8300168 |
|
BLAKE2b-256 | b7070c8361be2b0d00f983ef1c58e42f05b742b868d7635b4c186dfc83ca4cf0 |
File details
Details for the file torchlure-0.2405.2-py3-none-any.whl
.
File metadata
- Download URL: torchlure-0.2405.2-py3-none-any.whl
- Upload date:
- Size: 10.8 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.2 CPython/3.12.2
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | 7332eb79c1480557ac0adc4e2f611591b7536a5ec4d8bf8dfa96d16b3798c8ee |
|
MD5 | c06ccfb87f505a83a7d8e8693d1008c3 |
|
BLAKE2b-256 | 692c67ebfbd94d35308fc2c65bc06552ee4070a36d62d78983e890e6f25d40e6 |