diff --git a/README.md b/README.md index 587a97c71..813e72b8f 100644 --- a/README.md +++ b/README.md @@ -197,8 +197,8 @@ from tape.datasets import TRRosettaDataset # Download data and place it under `/trrosetta` -train_data = TRRosettaDatset('', 'train') # will subsample MSAs -valid_data = TRRosettaDatset('', 'valid') # will not subsample MSAs +train_data = TRRosettaDataset('', 'train') # will subsample MSAs +valid_data = TRRosettaDataset('', 'valid') # will not subsample MSAs model = TRRosetta.from_pretrained('xaa') # valid choices are 'xaa', 'xab', 'xac', 'xad', 'xae'. Each corresponds to one of the ensemble models.