velocitatem/PHANTOM
1
0
Fork 0
mirror of https://github.com/velocitatem/PHANTOM.git synced 2026-05-31 08:33:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

updaing coi definition

Browse Source
This commit is contained in:
Daniel Rosel
2026-02-05 12:47:13 +01:00
parent ebd2378859
commit e44feb7da0
3 changed files with 27 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
engine/train.py
View File

@@ -31,7 +31,7 @@ model.save("phantom_sac")
wandb.finish()
# test trained policy
env = PHANTOM(n_products=10, alpha=0.3, render_mode="human")
env = PHANTOM(n_products=10, alpha=0.3, render_mode=None)
obs, _ = env.reset()
for _ in range(100):
action, _ = model.predict(obs, deterministic=True)