Skip to content

Commit 35a6656

Browse files
committed
utils: Refactor to reformat code
1 parent e032682 commit 35a6656

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

utils.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -31,10 +31,10 @@ def get_model(config: dict, pretrained=False) -> torch.nn.Module:
3131

3232
if config['model'] == 'UNet':
3333
model = models.unet.UNet(config[config['model']]['num_classes'])
34-
elif config['model'] == 'Proposed':
35-
model = models.proposed.Proposed(config[config['model']]['num_classes'])
3634
elif config['model'] == 'Backbone':
3735
model = models.backbone.Backbone(config[config['model']]['num_classes'])
36+
elif config['model'] == 'Proposed':
37+
model = models.proposed.Proposed(config[config['model']]['num_classes'])
3838
else:
3939
raise NameError('Wrong model_name.')
4040

@@ -46,7 +46,6 @@ def get_model(config: dict, pretrained=False) -> torch.nn.Module:
4646
return model
4747

4848

49-
# Cityscapes 데이터셋
5049
class Cityscapes:
5150
def __init__(self, config: dict):
5251
self.config = config

0 commit comments

Comments
 (0)