For the stack experiments: ./main.py --task=stack ./main.py --task=stack --stack_fraction_values_for_train=0.75 ./main.py --task=stack --stack_fraction_values_for_train=0.75 --stack_nb_stacks=3 Each takes ~1h10min on a 4090.