摘要: 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)