We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2b6138b commit 55d14ceCopy full SHA for 55d14ce
pysipfenn/core/pysipfenn.py
@@ -354,7 +354,7 @@ def makePredictions(self,
354
t0 = perf_counter()
355
model = models[net]
356
model.eval()
357
- if 'OnnxDropoutDynamic()' in {str(module) for module in list(m._modules.values())}:
+ if 'OnnxDropoutDynamic()' in {str(module) for module in list(model._modules.values())}:
358
tempOut = model(dataIn, None)
359
else:
360
tempOut = model(dataIn)
0 commit comments