소스 검색

sphinx style guide

justheuristic 5 년 전
부모
커밋
9e9f8ff330
1개의 변경된 파일1개의 추가작업 그리고 0개의 파일을 삭제
  1. 1 0
      tesseract/client/gating_function.py

+ 1 - 0
tesseract/client/gating_function.py

@@ -16,6 +16,7 @@ class GatingFunction(nn.Module):
                  k_best, k_min=1, timeout_after_k_min=1.0, uid_prefix='', expert_padding=None):
         """
         A torch module that selects experts across the network and averages their predictions
+
         :param in_features: common input size for experts and gating function
         :param grid_size: tesseract dimensions that form expert uid (see below)
         :param uid_prefix: common prefix for all expert uids