Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -197,8 +197,8 @@ from tape.datasets import TRRosettaDataset

# Download data and place it under `<data_path>/trrosetta`

train_data = TRRosettaDatset('<data_path>', 'train') # will subsample MSAs
valid_data = TRRosettaDatset('<data_path>', 'valid') # will not subsample MSAs
train_data = TRRosettaDataset('<data_path>', 'train') # will subsample MSAs
valid_data = TRRosettaDataset('<data_path>', '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.

Expand Down