We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
weights_only=False
1 parent fd19f8a commit c108ea9Copy full SHA for c108ea9
auto3dseg/algorithm_templates/dints/configs/network.yaml
@@ -1,7 +1,7 @@
1
---
2
training_network:
3
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)"
+ arch_ckpt: "$torch.load(@training_network#arch_ckpt_path, map_location=torch.device('cuda'), weights_only=False)"
5
dints_space:
6
_target_: TopologyInstance
7
channel_mul: 1
0 commit comments