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):
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.3.dev0.tar.gz
(11.8 kB
view details)
Built Distribution
File details
Details for the file torchlure-0.2405.3.dev0.tar.gz.
File metadata
- Download URL: torchlure-0.2405.3.dev0.tar.gz
- Upload date:
- Size: 11.8 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.2 CPython/3.12.2
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 7d23b0483586b48822fe847298231a96d6e72f86d76bbf456a2dbbb65db5f3b3 |
|
| MD5 | 151d8aa3f36e842b244445d69a6a3368 |
|
| BLAKE2b-256 | dfa547edd9711738a4281704c7271f602c1c62ab28c571fd5e2575272737487f |
File details
Details for the file torchlure-0.2405.3.dev0-py3-none-any.whl.
File metadata
- Download URL: torchlure-0.2405.3.dev0-py3-none-any.whl
- Upload date:
- Size: 12.9 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 | e750567a88d0af4a7066b84059e1a8f226b4b754d9b5ae34f90f2930899b58f1 |
|
| MD5 | 5dc088f2ada2d5b87d379d21924d9042 |
|
| BLAKE2b-256 | 4095f84f1b50ed6500c53fe74a7e326875ad9a9d2dbde5051c51835c6c284489 |
