Commit f19d3383 authored by Justin Borromeo's avatar Justin Borromeo
Browse files

Mobile only

parent 29e2bbc1
......@@ -451,12 +451,12 @@ class cma_approach(object):
self.k,
self.topology_file,
1,
(1/max_latency),
max_latency*self.k,
1/full_latency[full_max_idx],
full_latency[full_max_idx],
(1/max_latency)/(1/full_latency[full_max_idx]),
(max_latency*self.k)/full_latency[full_max_idx],
(1/max_latency), # tp partition
max_latency*self.k, # latency partition
1/full_latency[full_max_idx], # tp fullmap
full_latency[full_max_idx], # latency fm
(1/max_latency)/(1/full_latency[full_max_idx]), # tp gain
(max_latency*self.k)/full_latency[full_max_idx], # latency penalty
layer,
res,
self.end-self.start,
......
......@@ -6,14 +6,14 @@ import path_constant as pc
import cma_approach_square_size as ap
nets = [
('FasterRCNN', 46, 23),
('mobilenet', 27, 14),
('yolo_tiny', 10, 9),
('googlenet', 58, 29),
('alexnet', 8, 8),
('AlphaGoZero', 8, 8),
('ncf_rec', 8, 8),
('resnet_50_v1', 53, 27)
#('FasterRCNN', 46, 23),
('mobilenet', 27, 20),
#('yolo_tiny', 10, 9),
#('googlenet', 58, 29),
#('alexnet', 8, 8),
#('AlphaGoZero', 8, 8),
#('ncf_rec', 8, 8),
#('resnet_50_v1', 53, 27)
]
max_res_sizes = [1920*9*1]
......@@ -82,7 +82,7 @@ if __name__ == "__main__":
pool = Pool(processes = cpu_count() - 2)
for result in pool.imap_unordered(trial, approaches):
# This will overwrite the previous file.
with open(pc.RESULT_CSV_PATH+'cma_logmore_sq_full_evo_packing.csv', 'a') as output_csv:
with open(pc.RESULT_CSV_PATH+'cma_logmore_sq_full_evo_packing_mobile.csv', 'a') as output_csv:
for evo in result:
writer = csv.writer(output_csv, delimiter=',', lineterminator="\n")
writer.writerow(evo)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment