Commit 00e135ad authored by Richard Xue's avatar Richard Xue Committed by Gao, Xiang
Browse files

typo ensemble (#96)

parent 8f1cb0c7
......@@ -50,7 +50,7 @@ class ANIModel(torch.nn.ModuleList):
class Ensemble(torch.nn.ModuleList):
"""Compute the average output of an ensemeble of modules."""
"""Compute the average output of an ensemble of modules."""
def forward(self, species_input):
outputs = [x(species_input)[1] for x in self]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment