MnemoDyn / Orion_333 /hparams.yaml
vhluong's picture
Add MnemoDyn checkpoint
73d83e5 verified
raw
history blame contribute delete
688 Bytes
D: 333
D_out:
- 333
- 333
- 333
- 333
K_LC:
- 2
- 2
- 2
- 2
- 2
- 2
K_dense: 32
LC_rank: 4
conv_bias: true
dataset_type: GordonHCP
dense_rank: 8
dim_d: 150
dim_k: 350
duration: 1
epoch: 1000
exp: debug
h_rank: 5
huber_delta: 1.0
interpol: spline
linear_bias: true
loss_type: total
lr: 0.001
masked_modelling: false
model_pred_save_freq: 100
n_levels: 6
nb: 3
nonlinearity: tanh
normalize: robust_scaler
num_classes: 1
num_sparse_LC: 5
one_channel: -1
original_length: 1200
predict: false
profile: false
scale: false
seed: 34
storage: /nas/sourav/Result/Orion_333/
subset: true
test_bs: 16
train_bs: 8
use_mRLoss: false
use_mixed_precision: false
use_residual: true
valid_bs: 16
wave: db2