Exact code used for producing benchmarks

[grananqvist]

updating with the code I used to produce the benchmark results. The only difference now is project id for wandb and the group id env variable.