Skip to content
Snippets Groups Projects
Unverified Commit c2701467 authored by Ziyi Wu's avatar Ziyi Wu Committed by GitHub
Browse files

fix bug in test.py (#819)

parent 31ef559a
Branches
No related tags found
No related merge requests found
......@@ -181,7 +181,8 @@ def main():
# palette for visualization in segmentation tasks
if 'PALETTE' in checkpoint.get('meta', {}):
model.PALETTE = checkpoint['meta']['PALETTE']
else:
elif hasattr(dataset, 'PALETTE'):
# segmentation dataset has `PALETTE` attribute
model.PALETTE = dataset.PALETTE
if not distributed:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment