diff --git a/src/fairchem/core/datasets/oc22_lmdb_dataset.py b/src/fairchem/core/datasets/oc22_lmdb_dataset.py index 15c627a5a..867a72726 100644 --- a/src/fairchem/core/datasets/oc22_lmdb_dataset.py +++ b/src/fairchem/core/datasets/oc22_lmdb_dataset.py @@ -116,11 +116,8 @@ def __init__(self, config, transform=None) -> None: if self.train_on_oc20_total_energies: with open(config["oc20_ref"], "rb") as fp: self.oc20_ref = pickle.load(fp) - self.subsample = aii(self.config.get("subsample", False), bool) def __len__(self) -> int: - if self.subsample: - return min(self.subsample, self.num_samples) return self.num_samples def __getitem__(self, idx):