diff --git a/Inference.py b/Inference.py index 1628029..5df1fff 100644 --- a/Inference.py +++ b/Inference.py @@ -91,7 +91,7 @@ def pushResult(self): global result string = '%4s %d = %7.2f ms (%6.2f ms)' % (self.dev_type.upper(), self.idx, \ self.result_time.exec_time + self.result_time.io_time, self.result_time.io_time) - if self.predict_time > 0: + if self.predict_time is not None: string += ' | %7.2f ms' % (self.predict_time) string += ' [%3d]\n' % (self.batch_size) result += string diff --git a/Partition.py b/Partition.py index a834360..b1c99ea 100644 --- a/Partition.py +++ b/Partition.py @@ -127,12 +127,12 @@ def estimateDevTime(self, dev, batch_size): yp_exec = [dev_perf[xp[0]].exec_time, dev_perf[xp[1]].exec_time] intp_time = np.interp(batch_size, xp, yp_exec) * batch_size + intp_io_time = 0. if self.with_io_time: yp_io = [dev_perf[xp[0]].io_time, dev_perf[xp[1]].io_time] intp_io_time = np.interp(batch_size, xp, yp_io) * batch_size - intp_time += intp_io_time - return intp_time + return intp_time + intp_io_time def getMaxDiffDev(self): ret_dev = None diff --git a/table_densenet-121 b/table_densenet-121 new file mode 100644 index 0000000..8bdbdb8 Binary files /dev/null and b/table_densenet-121 differ diff --git a/table_inception_v3 b/table_inception_v3 new file mode 100644 index 0000000..f9f1041 Binary files /dev/null and b/table_inception_v3 differ diff --git a/table_resnet-18 b/table_resnet-18 new file mode 100644 index 0000000..7078e7d Binary files /dev/null and b/table_resnet-18 differ diff --git a/table_resnet-34 b/table_resnet-34 new file mode 100644 index 0000000..9fee097 Binary files /dev/null and b/table_resnet-34 differ diff --git a/table_resnet-50 b/table_resnet-50 new file mode 100644 index 0000000..008516f Binary files /dev/null and b/table_resnet-50 differ diff --git a/table_squeezenet_v1.0 b/table_squeezenet_v1.0 new file mode 100644 index 0000000..09f7fed Binary files /dev/null and b/table_squeezenet_v1.0 differ diff --git a/table_squeezenet_v1.1 b/table_squeezenet_v1.1 new file mode 100644 index 0000000..d19206c Binary files /dev/null and b/table_squeezenet_v1.1 differ