RuntimeError: CUDA out of memory. Tried to allocate… but memory is empty

[Pierre Champion]

I was constantly getting a out of memory error even through I reduced my minibatch size.

I've replace

pkwrap/pkwrap/chain.py

Line 687 in e055fbc

 minibatch_size="1:64", # TODO: this should come from a config 

and

pkwrap/pkwrap/chain.py

Line 519 in e055fbc

 minibatch_size="1:64", 

to

minibatch_size=f"1:{chain_opts.minibatch_size.split(",")[-1]}", 

Is this correct ?
Thanks for this toolkit!

[Srikanth M R]

Right, that would work too. But before that I would test with a small minibatch size like 8 or 16. That should resolve your error, imo.

[Pierre Champion]

thanks!