Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
# @package _global_ | |
defaults: | |
- override /data: dti_data | |
- override /model: dti_model | |
- override /trainer: gpu | |
seed: 12345 | |
trainer: | |
min_epochs: 1 | |
max_epochs: 500 | |
precision: 16-mixed | |
callbacks: | |
early_stopping: | |
patience: 50 | |
data: | |
num_workers: 8 | |