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 *
DEFALUT_CANDIDATES = [
ShearXY,
TranslateXY,
Rotate,
AutoContrast,
Invert,
# Rotate,
# AutoContrast,
# Invert,
Equalize,
Solarize,
Posterize,
Contrast,
Color,
# Color,
Brightness,
Sharpness,
Cutout,
......@@ -87,7 +87,7 @@ def validate_child(args, model, dataset, subset_indx, transform, device=None):
dataset.transform = transform
subset = Subset(dataset, subset_indx)
data_loader = get_dataloader(args, subset, pin_memory=False)
data_loader = get_dataloader(args, subset, pin_memory=False) ##
return validate(args, model, criterion, data_loader, 0, None, device)
......
{"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
04:28
\ No newline at end of file
{"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
{"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
This diff is collapsed. Click to expand it.
This diff is collapsed. Click to expand it.
max step = 3000
step = 500
lr = 0.0001
batch_size = 64
DEFALUT_CANDIDATES = [
ShearXY,
TranslateXY,
# Rotate,
# AutoContrast, #߰
# Invert,
Equalize,
Solarize,
Posterize,
Contrast,
# Color,
Brightness,
Sharpness,
Cutout,
# SamplePairing,
]
\ No newline at end of file