Skip to content
Snippets Groups Projects
Commit 70449393 authored by hsfzxjy's avatar hsfzxjy
Browse files

fix bug

parent ff79976c
No related branches found
No related tags found
No related merge requests found
......@@ -261,6 +261,7 @@ blocks_dict = {
class HighResolutionNet(nn.Module):
def __init__(self, config, **kwargs):
global ALIGN_CORNERS
extra = config.MODEL.EXTRA
super(HighResolutionNet, self).__init__()
ALIGN_CORNERS = config.MODEL.ALIGN_CORNERS
......
......@@ -414,6 +414,7 @@ blocks_dict = {
class HighResolutionNet(nn.Module):
def __init__(self, config, **kwargs):
global ALIGN_CORNERS
extra = config.MODEL.EXTRA
super(HighResolutionNet, self).__init__()
ALIGN_CORNERS = config.MODEL.ALIGN_CORNERS
......
......@@ -141,7 +141,7 @@ def main():
drop_last=True,
sampler=train_sampler)
extra_epoch_iters = 0
if config.DATASET.EXTRA_TRAIN_SET:
extra_train_dataset = eval('datasets.'+config.DATASET.DATASET)(
root=config.DATASET.ROOT,
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment