jbm119's picture
Upload train_args.yaml with huggingface_hub
87836f5 verified
Raw
History Blame Contribute Delete
2.69 kB
aa: null
amp: false
amp_dtype: float16
amp_impl: native
aug_repeats: 0
aug_splits: 0
bandpass_freq:
- 0.01
- 150
batch_size: 384
bce_loss: true
bce_pos_weight: null
bce_sum: false
bce_target_thresh: null
bn_eps: null
bn_momentum: null
channels_last: false
checkpoint_hist: 3
class_map: ''
clip_grad: null
clip_mode: norm
color_jitter: 0.4
color_jitter_prob: null
cooldown_epochs: 0
crop_pct: null
cutmix: 0.0
cutmix_minmax: null
data: null
data_dir: /home/jmoody/ecgflow/data/mimic-iv-ecg
dataset: ecgflow/mimic
dataset_download: false
decay_epochs: 90
decay_milestones:
- 90
- 180
- 270
decay_rate: 0.1
dist_bn: reduce
drop: 0.0
drop_block: null
drop_connect: null
drop_path: null
early_stop: null
epoch_repeats: 0.0
epochs: 300
eval_metric: top1
experiment: mtsm-p50-d12-h8
fast_norm: false
filter_wf: false
ft_head: false
ft_top: 0
fuser: ''
gaussian_blur_prob: null
gp: null
grad_accum_steps: 1
grad_checkpointing: false
grayscale_prob: null
head_init_bias: null
head_init_scale: null
hflip: 0.5
img_size: null
in_chans: null
initial_checkpoint: ''
input_img_mode: null
input_key: null
input_size:
- 1
- 8
- 5000
interpolation: ''
jsd_loss: false
keep_head: false
layer_decay: null
local_rank: 0
log_interval: 50
log_wandb: false
lr: 0.001
lr_base: 0.1
lr_base_scale: ''
lr_base_size: 256
lr_cycle_decay: 0.5
lr_cycle_limit: 1
lr_cycle_mul: 1.0
lr_k_decay: 7.0
lr_noise: null
lr_noise_pct: 0.67
lr_noise_std: 1.0
mean: null
min_lr: 0
mixup: 0.0
mixup_mode: batch
mixup_off_epoch: 0
mixup_prob: 1.0
mixup_switch_prob: 0.5
model: mtsm_base_patch50
model_ema: false
model_ema_decay: 0.9998
model_ema_force_cpu: false
model_kwargs:
img_size: 5000
momentum: 0.9
mse_loss: false
mslq_loss: false
no_aug: true
no_ddp_bb: false
no_prefetcher: true
no_resume_opt: false
notch_freq: 60.0
num_classes: -1
opt: adamw
opt_betas: null
opt_eps: null
opt_kwargs: {}
output: /home/jmoody/ecgflow/experiments//mimic
patience_epochs: 10
pin_mem: false
pretrained: false
pretrained_path: null
ratio:
- 0.75
- 1.3333333333333333
recount: 1
recovery_interval: 0
remode: pixel
reprob: 0.0
resplit: false
resume: ''
save_images: false
scale:
- 0.08
- 1.0
scale_y: false
sched: cosine
sched_on_updates: true
seed: 42
smoothing: 0.01
split_bn: false
ssl: true
ssl_mask_ratio: 0.6
start_epoch: null
std: null
sync_bn: false
synchronize_step: false
target_key: null
torchcompile: null
torchscript: false
train_interpolation: random
train_num_samples: null
train_split: train
tta: 0
use_multi_epochs_loader: false
val_num_samples: null
val_split: validation
validation_batch_size: null
vflip: 0.0
warmup_epochs: 5
warmup_lr: 1.0e-05
warmup_prefix: false
weight_decay: 2.0e-05
worker_seeding: all
workers: 12