Brush dies with Killed

[poly2it]

After some time, the latest version (f72f518) of Brush dies with Killed:

$ RUST_BACKTRACE=1 brush_app
Killed

There seem to also be some internal sufferings:

$ brush_app
thread 'tokio-runtime-worker' panicked at /build/cargo-vendor-dir/burn-jit-0.17.0/src/kernel/conv/conv2d/im2col.rs:218:10:
Image too large to run even one batch at once
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace
thread 'tokio-runtime-worker' panicked at /build/cargo-vendor-dir/burn-jit-0.17.0/src/kernel/conv/conv2d/col2im.rs:65:10:
Image too large to run even one batch at once
Killed

However, those don't kill the app instantly.

Linux (NixOS) with an NVIDIA 3060.

[poly2it]

I can reproduce this on 0.2.0.

[poly2it]

The error is not present on ca1caf7.