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:
|
| 3 |
-
size
|
|
|
|
| 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: /
|
| 164 |
wandb_entity: biomed-signal-processing
|
| 165 |
-
wandb_dir: /
|
| 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:
|
| 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
|