51 lines
1.1 KiB
YAML
51 lines
1.1 KiB
YAML
|
run:
|
||
|
|
num_samples: 1
|
||
|
|
resources_per_trial:
|
||
|
|
cpu: 128
|
||
|
|
gpu: 1
|
||
|
|
scheduler:
|
||
|
|
_target_: ray.tune.schedulers.ASHAScheduler
|
||
|
|
metric: val_loss
|
||
|
|
mode: min
|
||
|
|
max_t: 20000
|
||
|
|
grace_period: 1
|
||
|
|
reduction_factor: 2
|
||
|
|
config:
|
||
|
|
choose_frame_spatial:
|
||
|
|
_target_: ray.tune.grid_search
|
||
|
|
values:
|
||
|
|
- _target_: data.choose_frame_spatial.mean.ChooseFrameSpatial
|
||
|
|
interval: [-3,0]
|
||
|
|
|
||
|
|
|
||
|
|
features_scaling:
|
||
|
|
_target_: ray.tune.grid_search
|
||
|
|
values:
|
||
|
|
- _target_: data.features_scaling.max_min.FeatureScaling
|
||
|
|
|
||
|
|
|
||
|
|
labels_scaling:
|
||
|
|
_target_: ray.tune.grid_search
|
||
|
|
values:
|
||
|
|
- _target_: data.labels_scaling.max_min.LabelScaling
|
||
|
|
|
||
|
|
model:
|
||
|
|
_target_: ray.tune.grid_search
|
||
|
|
values:
|
||
|
|
# - _target_: model.FCNN.FCNN
|
||
|
|
# - _target_: model.CNN_LSTM_FCNN.CNN_LSTM_FCNN
|
||
|
|
- _target_: model.CNN1D_FCNN.CNN1D_FCNN
|
||
|
|
criterion:
|
||
|
|
_target_: ray.tune.grid_search
|
||
|
|
values:
|
||
|
|
- _target_: torch.nn.MSELoss
|
||
|
|
optimizer:
|
||
|
|
_target_: ray.tune.grid_search
|
||
|
|
values:
|
||
|
|
- _target_: torch.optim.Adam
|
||
|
|
_partial_: true
|
||
|
|
lr: 0.0001
|
||
|
|
batch_size:
|
||
|
|
_target_: ray.tune.grid_search
|
||
|
|
values:
|
||
|
|
- 128
|