Commit fcc32e83 authored by LongChan's avatar LongChan
Browse files

AGO square dataset

parent 6bc1f282
This diff is collapsed.
......@@ -27,12 +27,13 @@ def organize(cycle_csv_path, detail_csv_path):
if __name__ == "__main__":
path = sys.argv[1]
runname = sys.argv[2]
print(sys.argv)
out_path = sys.argv[3]
maxsize = int(sys.argv[4])
square = bool(sys.argv[5])
frames = []
dimension_found = []
if square:
for i in range(maxsize):
demension = str(i+1) + "x" + str(i+1)
......
#!/bin/bash
# ./generate_final_square_csv.sh yolo_tiny ../optimization_algo/yolo_tiny_mem_bound.csv 960
# ./generate_final_square_csv.sh AlphaGoZero ../optimization_algo/data_source/AlphaGoZero_square_mem_bound.csv 960
if [ "$1" != "" ]; then
# echo $1
# echo $2
# echo $3
python3 csv_reorganizer.py outputs/ $1 $2 $3 True
else
echo "Positional parameter 1 is empty"
......
......@@ -279,7 +279,7 @@ class scale:
print(all_arr_dim_list)
pool = Pool(processes = 8)
pool = Pool(processes = 12)
for pro in pool.imap_unordered(self.run_mp_once, all_arr_dim_list):
self.run_name = net_name + "_" + self.dataflow + "_" + str(pro[0]) + "x" + str(pro[1])
self.cleanup(pro)
......
Supports Markdown
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