xueh commited on
Commit
2e42c3d
·
verified ·
1 Parent(s): c8017d3

Upload 2 files

Browse files
small/snraware_small_model.pts CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:672d9583e85ec9f418695fdf73180daed47b885bc9804291d9dc0ad02219ad91
3
- size 118325006
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:08a78f0e61e7657356edfa6f37f1e3e97b6fa380fe41a357584e05123d7caed5
3
+ size 118324558
small/snraware_small_model.yaml CHANGED
@@ -160,9 +160,9 @@ trainer:
160
  logging:
161
  project: ifm-mri-denoising
162
  run_name: ifm-mri-denoising-small-no-epoch
163
- output_dir: /data1/inference_time/models/small
164
  wandb_entity: biomed-signal-processing
165
- wandb_dir: /data1/inference_time/models/small/wandb
166
  use_wandb: true
167
  save_ckpt_every_n_epochs: 16
168
  save_batches_to_output_dir: false
@@ -180,7 +180,7 @@ step_mode:
180
  steps_to_save_ckpts: 5000
181
  train_data_dir: /data/imaging-fm-projects/mri/denoising/data/whole/tra
182
  test_data_dir: /data/imaging-fm-projects/mri/denoising/data/whole/test
183
- seed: null
184
  max_epochs: 320
185
  batch_size: 4
186
  num_workers: 4
@@ -205,3 +205,11 @@ overlap_for_inference:
205
  device_mesh:
206
  - 8
207
  - 2
 
 
 
 
 
 
 
 
 
160
  logging:
161
  project: ifm-mri-denoising
162
  run_name: ifm-mri-denoising-small-no-epoch
163
+ output_dir: /data/models/ifm-mri-denoising_20251011_171331/ifm-mri-denoising-small-no-epoch/checkpoints
164
  wandb_entity: biomed-signal-processing
165
+ wandb_dir: /data/ifm-mri-denoising/ifm-mri-denoising_20251011_171331/ifm-mri-denoising-small-no-epoch/wandb
166
  use_wandb: true
167
  save_ckpt_every_n_epochs: 16
168
  save_batches_to_output_dir: false
 
180
  steps_to_save_ckpts: 5000
181
  train_data_dir: /data/imaging-fm-projects/mri/denoising/data/whole/tra
182
  test_data_dir: /data/imaging-fm-projects/mri/denoising/data/whole/test
183
+ seed: 1595086465
184
  max_epochs: 320
185
  batch_size: 4
186
  num_workers: 4
 
205
  device_mesh:
206
  - 8
207
  - 2
208
+ date: rank_0_200600_17-13-51-20251011
209
+ rank: 0
210
+ global_rank: 0
211
+ rank_str: "\e[34m\e[47mrank 0, global_rank 0 \e[0m"
212
+ device: cuda:0
213
+ yaml_file: null
214
+ log_file_name: null
215
+ ddp: true