SEMS-model-training/configs/main_train.yaml
2024-09-11 19:15:44 +08:00

21 lines
425 B
YAML

defaults:
- _self_
- hydra: default
- ray_tune: default
task_name: main
raw_spectral_data_dir: /code/admin/20240806-NanEr-5-8-data/rawdata
raw_labels_dir: /code/admin/20240806-NanEr-5-8-data/labels/NanEr
dataset_dir: /code/admin/20240806-NanEr-5-8-data/dataset
labels_name: ["TSC_T","TSC_C","TSC_C_lab","TSC_P_lab"]
#是
dataset:
train_ratio: 0.8
validate_ratio: 0.1
num_worker: 128
train:
max_epoch: 10000