소스 검색

remove debugprint

justheuristic 5 년 전
부모
커밋
cb1909a3bb
1개의 변경된 파일0개의 추가작업 그리고 1개의 파일을 삭제
  1. 0 1
      tests/test_moe.py

+ 0 - 1
tests/test_moe.py

@@ -63,7 +63,6 @@ def test_compute_expert_scores():
 
         for b in range(len(ii)):
             for e in range(len(ii[b])):
-                print(end='.')
                 assert torch.allclose(logits[b, e], gx[b, ii[b][e]] + gy[b, jj[b][e]]), "compute_expert_scores returned incorrect score"
 
         dht.shutdown()