mirror of
https://github.com/AntoineHX/BU_Stoch_pool.git
synced 2025-05-04 01:30:48 +02:00
fixed problem with inference with MyLeNetNoceil
This commit is contained in:
parent
019310e60c
commit
5504ca67ab
4 changed files with 157 additions and 138 deletions
|
@ -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)))
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue