suhyeonlee / WildNet

Official PyTorch Implementation of WildNet: Learning Domain Generalized Semantic Segmentation from the Wild, CVPR 2022

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

wild_loader seems to have some problem

Tracy-git opened this issue · comments

thks your excellent work,and when i run this repo,i only use cityscapes datasets , an error occurred ,
error appears on line 540 to 548 in dataset/__init.

` for i, wild_set in enumerate(wild_sets):
if args.syncbn:
wild_sampler = DistributedSampler(wild_set, pad=True, permutation=True, consecutive_sample=False)
else:
wild_sampler = None
wild_loader = DataLoader(wild_set, batch_size=args.train_batch_size,
num_workers=args.num_workers, pin_memory=True, shuffle=(wild_sampler is None), drop_last=True, sampler = wild_sampler)

return train_loader, val_loaders, wild_loader, train_set, extra_val_loader`

UnboundLocalError: local variable 'wild_loader' referenced before assignment