Skip to content

Commit c108ea9

Browse files
authored
Update weights_only=False in dints algorithm (#413)
Fix #411 Update to use `weights_only=False` in dints since the checkpoint not only save weights https://github.com/KumoLiu/research-contributions/blob/main/auto3dseg/algorithm_templates/dints/configs/network.yaml#L12 Signed-off-by: YunLiu <[email protected]>
1 parent fd19f8a commit c108ea9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

auto3dseg/algorithm_templates/dints/configs/network.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
---
22
training_network:
33
arch_ckpt_path: "$@bundle_root + '/scripts/arch_code.pth'"
4-
arch_ckpt: "$torch.load(@training_network#arch_ckpt_path, map_location=torch.device('cuda'), weights_only=True)"
4+
arch_ckpt: "$torch.load(@training_network#arch_ckpt_path, map_location=torch.device('cuda'), weights_only=False)"
55
dints_space:
66
_target_: TopologyInstance
77
channel_mul: 1

0 commit comments

Comments
 (0)