2020-1-capstone-design2 / 2016104167

Go to a project
Toggle navigation Toggle navigation pinning
조현아
cf89600f 1 parent bba51889

rm a few aug policies

Expand all
Inline Side-by-side
Showing 18 changed files with 34 additions and 5 deletions
...@@ -17,14 +17,14 @@ from utils import * ...@@ -17,14 +17,14 @@ from utils import *
17 DEFALUT_CANDIDATES = [ 17 DEFALUT_CANDIDATES = [
18 ShearXY, 18 ShearXY,
19 TranslateXY, 19 TranslateXY,
20 - Rotate, 20 + # Rotate,
21 - AutoContrast, 21 + # AutoContrast,
22 - Invert, 22 + # Invert,
23 Equalize, 23 Equalize,
24 Solarize, 24 Solarize,
25 Posterize, 25 Posterize,
26 Contrast, 26 Contrast,
27 - Color, 27 + # Color,
28 Brightness, 28 Brightness,
29 Sharpness, 29 Sharpness,
30 Cutout, 30 Cutout,
...@@ -87,7 +87,7 @@ def validate_child(args, model, dataset, subset_indx, transform, device=None): ...@@ -87,7 +87,7 @@ def validate_child(args, model, dataset, subset_indx, transform, device=None):
87 87
88 dataset.transform = transform 88 dataset.transform = transform
89 subset = Subset(dataset, subset_indx) 89 subset = Subset(dataset, subset_indx)
90 - data_loader = get_dataloader(args, subset, pin_memory=False) 90 + data_loader = get_dataloader(args, subset, pin_memory=False) ##
91 91
92 return validate(args, model, criterion, data_loader, 0, None, device) 92 return validate(args, model, criterion, data_loader, 0, None, device)
93 93
......
1 +{"use_cuda": true, "network": "resnet50", "dataset": "BraTS", "optimizer": "adam", "fast_auto_augment": true, "learning_rate": 0.01, "seed": null, "num_workers": 4, "print_step": 500, "val_step": 500, "scheduler": "exp", "batch_size": 128, "start_step": 0, "max_step": 10000, "augment_path": null}
...\ No newline at end of file ...\ No newline at end of file
1 +
2 +04:28
...\ No newline at end of file ...\ No newline at end of file
1 +{"use_cuda": true, "network": "resnet50", "dataset": "BraTS", "optimizer": "adam", "fast_auto_augment": true, "learning_rate": 0.0001, "seed": null, "num_workers": 4, "print_step": 500, "val_step": 500, "scheduler": "exp", "batch_size": 128, "start_step": 0, "max_step": 10000, "augment_path": null}
...\ No newline at end of file ...\ No newline at end of file
1 +{"use_cuda": true, "network": "resnet50", "dataset": "BraTS", "optimizer": "adam", "fast_auto_augment": true, "print_step": 300, "batch_size": 4, "val_step": 300, "max_step": 300, "learning_rate": 0.0001, "seed": null, "num_workers": 4, "scheduler": "exp", "start_step": 0, "augment_path": null}
...\ No newline at end of file ...\ No newline at end of file
This diff is collapsed. Click to expand it.
This diff is collapsed. Click to expand it.
1 +max step = 3000
2 +step = 500
3 +lr = 0.0001
4 +batch_size = 64
5 +
6 +
7 +DEFALUT_CANDIDATES = [
8 + ShearXY,
9 + TranslateXY,
10 + # Rotate,
11 + # AutoContrast, #߰
12 + # Invert,
13 + Equalize,
14 + Solarize,
15 + Posterize,
16 + Contrast,
17 + # Color,
18 + Brightness,
19 + Sharpness,
20 + Cutout,
21 +# SamplePairing,
22 +]
...\ No newline at end of file ...\ No newline at end of file