qninhdt
commited on
Commit
•
7e9239e
1
Parent(s):
8cc0674
cc
Browse files- .project-root +0 -0
- configs/trainer/default.yaml +1 -1
- swim/data/swim_data.py +1 -1
- train_autoencoder.sh +5 -0
.project-root
ADDED
|
File without changes
|
configs/trainer/default.yaml
CHANGED
|
@@ -9,7 +9,7 @@ accelerator: cpu
|
|
| 9 |
devices: 1
|
| 10 |
|
| 11 |
# mixed precision for extra speed-up
|
| 12 |
-
|
| 13 |
|
| 14 |
# perform a validation loop every N training epochs
|
| 15 |
check_val_every_n_epoch: 1
|
|
|
|
| 9 |
devices: 1
|
| 10 |
|
| 11 |
# mixed precision for extra speed-up
|
| 12 |
+
precision: 16
|
| 13 |
|
| 14 |
# perform a validation loop every N training epochs
|
| 15 |
check_val_every_n_epoch: 1
|
swim/data/swim_data.py
CHANGED
|
@@ -52,7 +52,7 @@ class SwimDataset(Dataset):
|
|
| 52 |
]
|
| 53 |
|
| 54 |
def __len__(self):
|
| 55 |
-
return len(self.data)
|
| 56 |
|
| 57 |
def __getitem__(self, idx):
|
| 58 |
data = self.data[idx]
|
|
|
|
| 52 |
]
|
| 53 |
|
| 54 |
def __len__(self):
|
| 55 |
+
return len(self.data)
|
| 56 |
|
| 57 |
def __getitem__(self, idx):
|
| 58 |
data = self.data[idx]
|
train_autoencoder.sh
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
python swim/train.py \
|
| 2 |
+
data.root_dir=/cm/shared/ninhnq3/datasets/swim_data \
|
| 3 |
+
data.img_size=512 \
|
| 4 |
+
data.batch_size=2 \
|
| 5 |
+
callbacks.model_checkpoint.dirpath=/cm/shared/ninhnq3/checkpoints/swim/autoencoder/simple
|