瀏覽代碼

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):
                  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
         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 in_features: common input size for experts and gating function
         :param grid_size: tesseract dimensions that form expert uid (see below)
         :param grid_size: tesseract dimensions that form expert uid (see below)
         :param uid_prefix: common prefix for all expert uids
         :param uid_prefix: common prefix for all expert uids