Time - ms
Mem - MB
Size - MB
from optimizer import Optimizer
Optimizer(
dep_filename="dep.csv",
prof_filenames=[
"prof.csv",
"prof.csv",
"prof.csv",
"prof.csv",
"prof.csv",
],
bandwidth=2000,
parallel=True,
ignore_latency=True,
iterations=1,
dir="testcase/explore"
)- This csv file contains the dependency relation between layers of a network.
- It has two columns: source, destination. Every entry represents an edge in the network.
- This csv file contains the profiling result of every layer on a particular device.
- One entry in this list represents one device. For example, in the basic optimizer block above, there are five devices available.
- It has five columns: layer_name, time,cpu_mem, cuda_mem, size, MACs
- The bandwidth of communication network between drones. The unit is MBps.
- data-computation parallel.
- Whether to ignore transfer latency. Mainly for testing.
Please use opt_wrapper.py when optimizing the network. Make a copy of the prof and dep files to the root directory.
You may modify the following part as needed.
bandwidth = 400
ignore_latency = False
iteration = 5
prof_filenames = [
"prof.csv",
"prof.csv",
# "prof.csv",
# "prof.csv",
# "prof.csv",
# "prof.csv",
# "prof.csv",
# "prof.csv",
]
benchmark = 114.46748520400001 # agx
# ...