Skip to content

Commit c0df59c

Browse files
committed
Disable convert weights printing
1 parent 5bd52c3 commit c0df59c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

segmentation_models_pytorch_3d/utils/convert_weights.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33

44
import torch
55

6-
def convert_2d_weights_to_3d(state_dict, verbose=True):
6+
def convert_2d_weights_to_3d(state_dict, verbose=False):
77
layers = list(state_dict.keys())
88
for layer in layers:
99
if (

0 commit comments

Comments
 (0)