摘要:
def weight_init(m): # 初始化权重 # print(m) if isinstance(m, torch.nn.Conv3d): n = m.kernel_size[0] * m.kernel_size[1] * m.kernel_size[2] * m.out_channels 阅读全文
posted @ 2023-03-23 14:22
Truman001
阅读(53)
评论(0)
推荐(0)
摘要:
import onnx # 加载ONNX模型 model_path = "model.onnx" model = onnx.load(model_path) # 遍历模型的图结构,获取每一层的节点属性 for node in model.graph.node: # 输出节点名称和类型 print(f 阅读全文
posted @ 2023-03-23 14:20
Truman001
阅读(1147)
评论(0)
推荐(1)
摘要:
import random my_list = [1, 2, 3, 4, 5] random_number = random.choice(my_list) print(random_number) 阅读全文
posted @ 2023-03-23 14:18
Truman001
阅读(41)
评论(0)
推荐(0)
摘要:
import random my_list = [1, 2, 3, 4, 5] random_number = random.choice(my_list) print(random_number) 阅读全文
posted @ 2023-03-23 14:17
Truman001
阅读(43)
评论(0)
推荐(0)