Table 1.

Datasets and their partitioning.

	Partition size (simulations)
Name	TRAIN	VALID	TEST
OPM-22k	17,600	2,200	2,200
ECL-32k	28,682	3,585	3,585
OPM-163k	130,778	16,347	16,347