Separate the vram logic for training and fragbench