Updates the model and config
Browse files- artifacts/dr-model.ckpt +2 -2
- conf/config.yaml +1 -1
artifacts/dr-model.ckpt
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:824332e4a82a313dfcc9e40b6859fda6b6e7fcd54e4b17a1e16a1f13136c3c54
|
| 3 |
+
size 67651330
|
conf/config.yaml
CHANGED
|
@@ -6,7 +6,7 @@ val_csv_path: data/diabetic-retinopathy-dataset/val.csv
|
|
| 6 |
seed: 42
|
| 7 |
batch_size: 128
|
| 8 |
num_workers: 32
|
| 9 |
-
use_class_weighting:
|
| 10 |
use_weighted_sampler: false # class weighting and weighted sampler are mutually exclusive
|
| 11 |
model_name: "densenet169"
|
| 12 |
max_epochs: 20
|
|
|
|
| 6 |
seed: 42
|
| 7 |
batch_size: 128
|
| 8 |
num_workers: 32
|
| 9 |
+
use_class_weighting: true
|
| 10 |
use_weighted_sampler: false # class weighting and weighted sampler are mutually exclusive
|
| 11 |
model_name: "densenet169"
|
| 12 |
max_epochs: 20
|