Skip to content

Commit 910e18a

Browse files
chyomin06fracape
authored andcommitted
[fix] dict of devices for creating the dataloader
1 parent da442bd commit 910e18a

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

compressai_vision/run/eval_remote_inference.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,9 @@ def setup(conf: DictConfig) -> dict[str, Any]:
7474
configure_conf(conf)
7575

7676
vision_model = create_vision_model(conf.misc.device.nn_parts, conf.vision_model)
77-
dataloader = create_dataloader(conf.dataset, conf.misc.device, vision_model.cfg)
77+
dataloader = create_dataloader(
78+
conf.dataset, conf.misc.device.nn_parts, vision_model.cfg
79+
)
7880
evaluator = create_evaluator(
7981
conf.evaluator,
8082
conf.dataset.datacatalog,

0 commit comments

Comments
 (0)