Memory requirement

[hquan7395]

Hi,

Thanks for your work. I was trying to run your model using

HYDRA_FULL_ERROR=1 python run_training.py task=dvs-gesture

The final error was

jax.errors.JaxRuntimeError: RESOURCE_EXHAUSTED: Out of memory while trying to allocate 1.00GiB.

Assuming the only problem here is memory constraints, do you have an estimate of the memory required?