Model Interpretability for PyTorch. ... conda install captum -c pytorch ... return self.lin2(self.relu(self.lin1(input))) model = ToyModel() model.eval().
確定! 回上一頁