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