all experiments run on an RTX NVIDIA A6000