[RL] larger circuits use more GPU-MEM than smaller circuits

[Co1lin]

Larger circuits use more GPU-MEM than smaller circuits even though we only train on k-hop neighborhood.