Refine doc and fix bug for testing MedianstopAssessor (#1998)

3c833f37 · Guanghui Ren(任广辉) · GitHub · e8d253c0 · 3c833f37
Unverified Commit 3c833f37 authored Feb 07, 2020 by Guanghui Ren(任广辉) Committed by GitHub Feb 07, 2020
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

src/sdk/pynni/nni/medianstop_assessor/test.py src/sdk/pynni/nni/medianstop_assessor/test.py +3 -3

No files found.
--- a/src/sdk/pynni/nni/medianstop_assessor/test.py
+++ b/src/sdk/pynni/nni/medianstop_assessor/test.py
@@ -31,11 +31,11 @@ def test():
    #       [1,1,1,1,1,1,1,1,1,1],
    #       [1,1,1,1,1,1,1,1,1,1]]

-    assessor = MedianstopAssessor(FLAGS.start_step, FLAGS.optimize_mode)
-    for i in range(4):
+    assessor = MedianstopAssessor(FLAGS.optimize_mode, FLAGS.start_step)
+    for i in range(len(lcs)):
        #lc = []
        to_complete = True
-        for k in range(10):
+        for k in range(len(lcs[0])):
            #d = random.randint(i*100+0, i*100+100)
            #lc.append(d)
            ret = assessor.assess_trial(i, lcs[i][:k+1])