Skip to content

Conversation

@AntonioMirarchi
Copy link
Contributor

@AntonioMirarchi AntonioMirarchi commented Sep 19, 2024

This PR introduces a custom version of MDCATH while preserving the original functionality to ensure backward compatibility.

New Arguments:

  • source_file and file_basename: these args enable users to create and train with their custom versions of the MDCATH dataset.
  • numNoHAtoms by default is None, max number of non-hydrogen atoms in the protein structure.
  • Assertions have been added to ensure that automatic downloading is restricted solely to the original version of the MDCATH dataset.

These enhancements aim to provide users with greater flexibility in dataset handling and customization, aligning with diverse research needs.

@AntonioMirarchi
Copy link
Contributor Author

@stefdoerr @sef43 can you merge it?

@stefdoerr stefdoerr merged commit f013b80 into torchmd:main Jan 28, 2025
2 checks passed
@AntonioMirarchi AntonioMirarchi deleted the mdcath_noh branch January 28, 2025 15:27
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants