Traceback (most recent call last):
File "/home/teacher/zzw/OpenTrack/train_policy.py", line 166, in
train(tyro.cli(Args))
File "/home/teacher/zzw/OpenTrack/train_policy.py", line 155, in train
make_inference_fn, params, _ = train_fn(
^^^^^^^^^
File "/home/teacher/zzw/OpenTrack/src/learning/ppo/train_ppo.py", line 584, in train
(training_state, env_state, training_metrics) = training_epoch_with_timing(
^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/teacher/zzw/OpenTrack/src/learning/ppo/train_ppo.py", line 512, in training_epoch_with_timing
result = training_epoch(training_state, env_state, key)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
jaxlib.xla_extension.XlaRuntimeError: RESOURCE_EXHAUSTED: Out of memory while trying to allocate 11828198992 bytes.
