Commit
•
8afb176
1
Parent(s):
797d116
Reorganized files
Browse files- tag.sh → slurm/tag.sh +0 -0
- train_a100_x1.sh → slurm/train_a100_x1.sh +0 -0
- train_a100_x4.sh → slurm/train_a100_x4.sh +0 -0
- train_v100_x1.sh → slurm/train_v100_x1.sh +0 -0
- train_v100_x4.sh → slurm/train_v100_x4.sh +0 -0
- data.py → src/data.py +1 -1
- labelmap.py → src/labelmap.py +0 -0
- metrics.py → src/metrics.py +0 -0
- trainer.py → src/trainer.py +2 -2
- train.py +1 -1
tag.sh → slurm/tag.sh
RENAMED
|
File without changes
|
train_a100_x1.sh → slurm/train_a100_x1.sh
RENAMED
|
File without changes
|
train_a100_x4.sh → slurm/train_a100_x4.sh
RENAMED
|
File without changes
|
train_v100_x1.sh → slurm/train_v100_x1.sh
RENAMED
|
File without changes
|
train_v100_x4.sh → slurm/train_v100_x4.sh
RENAMED
|
File without changes
|
data.py → src/data.py
RENAMED
|
@@ -14,7 +14,7 @@ import torch.utils.data as data
|
|
| 14 |
from torch.utils.data import DataLoader
|
| 15 |
|
| 16 |
|
| 17 |
-
from labelmap import DR_LABELMAP
|
| 18 |
|
| 19 |
|
| 20 |
DataRecord = Tuple[Image.Image, int]
|
|
|
|
| 14 |
from torch.utils.data import DataLoader
|
| 15 |
|
| 16 |
|
| 17 |
+
from src.labelmap import DR_LABELMAP
|
| 18 |
|
| 19 |
|
| 20 |
DataRecord = Tuple[Image.Image, int]
|
labelmap.py → src/labelmap.py
RENAMED
|
File without changes
|
metrics.py → src/metrics.py
RENAMED
|
File without changes
|
trainer.py → src/trainer.py
RENAMED
|
@@ -25,8 +25,8 @@ from transformers import AdamW
|
|
| 25 |
from transformers import AutoImageProcessor, ResNetForImageClassification
|
| 26 |
import lightning as L
|
| 27 |
|
| 28 |
-
from data import RetinopathyDataset, Split
|
| 29 |
-
from metrics import Metrics
|
| 30 |
|
| 31 |
|
| 32 |
def worker_init_fn(worker_id: int) -> None:
|
|
|
|
| 25 |
from transformers import AutoImageProcessor, ResNetForImageClassification
|
| 26 |
import lightning as L
|
| 27 |
|
| 28 |
+
from src.data import RetinopathyDataset, Split
|
| 29 |
+
from src.metrics import Metrics
|
| 30 |
|
| 31 |
|
| 32 |
def worker_init_fn(worker_id: int) -> None:
|
train.py
CHANGED
|
@@ -7,7 +7,7 @@ from lightning import Trainer
|
|
| 7 |
from lightning.pytorch.loggers import TensorBoardLogger
|
| 8 |
from lightning.pytorch.callbacks import ModelSummary
|
| 9 |
|
| 10 |
-
from trainer import ViTLightningModule
|
| 11 |
|
| 12 |
|
| 13 |
def main():
|
|
|
|
| 7 |
from lightning.pytorch.loggers import TensorBoardLogger
|
| 8 |
from lightning.pytorch.callbacks import ModelSummary
|
| 9 |
|
| 10 |
+
from src.trainer import ViTLightningModule
|
| 11 |
|
| 12 |
|
| 13 |
def main():
|