model to be the torchbeast baseline, using only one GPU. Adjust the baseline to only use the new blstats.