fixed problem with inference with MyLeNetNoceil

This commit is contained in:
Marco Pedersoli 2020-06-30 11:56:51 -04:00
parent 019310e60c
commit 5504ca67ab
4 changed files with 157 additions and 138 deletions

View file

@ -23,8 +23,8 @@ if __name__ == "__main__":
accs.append(max([x["test_acc"] for x in data]))
taccs.append(max([x["train_acc"] for x in data]))
# aug_accs.append(data['Aug_Accuracy'][1])
# times.append(data['Time'][0])
# mem.append(data['Memory'][1])
#times.append(data['Time'][0])
#mem.append(data['Memory'][1])
# acc_idx = [x['acc'] for x in data['Log']].index(data['Accuracy'])
# f1_max.append(max(data['Log'][acc_idx]['f1'])*100)
@ -36,5 +36,5 @@ if __name__ == "__main__":
print("Acc train : %.2f ~ %.2f"%(np.mean(taccs), np.std(taccs)))
# print("Acc : %.2f ~ %.2f / Aug_Acc %d: %.2f ~ %.2f"%(np.mean(accs), np.std(accs), data['Aug_Accuracy'][0], np.mean(aug_accs), np.std(aug_accs)))
# print("F1 max : %.2f ~ %.2f / F1 min : %.2f ~ %.2f"%(np.mean(f1_max), np.std(f1_max), np.mean(f1_min), np.std(f1_min)))
# print("Time (h): %.2f ~ %.2f"%(np.mean(times)/3600, np.std(times)/3600))
# print("Mem (MB): %d ~ %d"%(np.mean(mem), np.std(mem)))
#print("Time (h): %.2f ~ %.2f"%(np.mean(times)/3600, np.std(times)/3600))
#print("Mem (MB): %d ~ %d"%(np.mean(mem), np.std(mem)))