Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
_target_: deepscreen.data.dti_datamodule.DTIdatamodule | |
defaults: | |
- _self_ | |
- split: null | |
- drug_featurizer: null | |
- protein_featurizer: null | |
task: ${task.task} | |
n_class: ${oc.select:task.task.n_class,null} | |
data_dir: ${paths.data_dir} | |
dataset_name: null | |
batch_size: 16 | |
train_val_test_split: [0.7, 0.1, 0.2] | |
num_workers: 0 | |
pin_memory: false | |
train: ${train} |