update

jzuern · May 9, 2023 · b1deeeb · b1deeeb
1 parent c27390a
commit b1deeeb
Show file tree

Hide file tree

Showing 6 changed files with 426 additions and 468 deletions.
diff --git a/cfg-rlgpu.yaml b/cfg-rlgpu.yaml
@@ -23,7  23,7 @@ model:
     dataparallel: True
     device: cuda
     batch_size: 4
-    batch_size_reg: 128
     batch_size_reg: 384
     lr: 1e-4
     beta_lo: 0.9
     beta_hi: 0.999