1
Fork 0

feat(PPOTrainer.py): added logging messages

This commit is contained in:
Robin Meersman 2026-04-06 13:56:54 +02:00
parent 84f6fbd76e
commit b97fc30d5d
5 changed files with 135 additions and 27 deletions

6
uv.lock generated
View file

@ -35,6 +35,9 @@ dependencies = [
]
[package.optional-dependencies]
analysis = [
{ name = "tensorboard" },
]
cuda = [
{ name = "jax", extra = ["cuda13"] },
]
@ -64,12 +67,13 @@ requires-dist = [
{ name = "protobuf", specifier = ">=5.0.0" },
{ name = "pyopengl", specifier = ">=3.1.10" },
{ name = "pyopengl-accelerate", specifier = ">=3.1.10" },
{ name = "tensorboard", marker = "extra == 'analysis'" },
{ name = "torch", specifier = ">=2.4.0" },
{ name = "tyro", specifier = ">=1.0.10" },
{ name = "wandb", specifier = "==0.24.2" },
{ name = "warp-lang" },
]
provides-extras = ["cuda"]
provides-extras = ["cuda", "analysis"]
[package.metadata.requires-dev]
dev = [