Skip to content

GitLab

"rust/git@developer.sourcefind.cn:change/sglang.git" did not exist on "499c85f1318d5ad914a599050bd3f616a28007e0"
Commit f53de2be authored by Claudio Michaelis's avatar Claudio Michaelis Committed by Kai Chen
Browse files

Fix robustness data pipeline (#1476)

parent 3dc9ddb7
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tools/test_robustness.py
View file @ f53de2be
...@@ -350,9 +350,9 @@ def main(): ...@@ -350,9 +350,9 @@ def main():
aggregated_results[corruptions[0]][0] aggregated_results[corruptions[0]][0]
continue continue
test_data_cfg = copy.deepcopy(cfg.data.test)
# assign corruption and severity # assign corruption and severity
if corruption_severity > 0: if corruption_severity > 0:
test_data_cfg = copy.deepcopy(cfg.data.test)
corruption_trans = dict( corruption_trans = dict(
type='Corrupt', type='Corrupt',
corruption=corruption, corruption=corruption,
...@@ -368,7 +368,7 @@ def main(): ...@@ -368,7 +368,7 @@ def main():
# build the dataloader # build the dataloader
# TODO: support multiple images per gpu # TODO: support multiple images per gpu
# (only minor changes are needed) # (only minor changes are needed)
dataset = build_dataset(cfg.data.test) dataset = build_dataset(test_data_cfg)
data_loader = build_dataloader( data_loader = build_dataloader(
dataset, dataset,
imgs_per_gpu=1, imgs_per_gpu=1,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment