Parcourir la source

sphinx style guide

justheuristic il y a 5 ans
Parent
commit
9e9f8ff330
1 fichiers modifiés avec 1 ajouts et 0 suppressions
  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