分区 #2
Description
首先感谢您的代码分享，在尝试运行后有一个问题想要请教，以inceptionV2为例，DSL算法得到的分区边集合model_partition_edge的长度并不一定为4，但在InceptionBlockV2.py"文件中，assert len(model_partition_edge) == 4就会报错，该如何解决呢
python edge_api.py -i 172.17.4.209 -p 9999 -d cpu -t inception_v2
monitor client get latency : 3704.888750114464 MB/s
short message , model type has been sent successfully
edge_latency_list : [107.84478958230466, 13.369368622079492, 26.299802120774984, 57.26995733566582, 7.406370993703604, 26.44088137894869, 42.77969708200544, 17.51779713667929, 26.273923716507852, 27.527293609455228, 8.349713776260614, 14.03235960751772, 11.371594085358083, 0.06040649022907019]
cloud_latency_list : [69.2196918418631, 13.138559181243181, 97.52658936195076, 87.32770101632923, 7.599794655106962, 37.64628728386015, 25.54533185902983, 23.401623335666955, 30.511609348468482, 2.8537589590996504, 3.5421702777966857, 3.650573710910976, 1.1728882789611816, 0.19014575518667698]
start construct graph for model...
graph_partition_edge : [('v9', 'v10'), ('v0', 'v1'), ('v5', 'v15')]
partition edges : [(9, 10), (0, 1), (5, 5)]
short message , partition strategy has been sent successfully
Traceback (most recent call last):
File "/home/dzg/pwy/DADS-main/edge_api.py", line 70, in
start_client(ip, port, x, model_type, upload_bandwidth, device)
File "/home/dzg/pwy/DADS-main/server_func.py", line 101, in start_client
edge_model, _ = inference_utils.model_partition(model, model_partition_edge)
File "/home/dzg/pwy/DADS-main/utils/inference_utils.py", line 271, in model_partition
return construct_edge_cloud_inception_block(model, model_partition_edge)
File "/home/dzg/pwy/DADS-main/models/InceptionBlockV2.py", line 265, in construct_edge_cloud_inception_block
assert len(model_partition_edge) == 4
AssertionError