Filter kwargs for preprocess_input (#71)

qubvel · web-flow · commit 91366f510127 · 2019-09-13T15:34:34.000+03:00
diff --git a/efficientnet/model.py b/efficientnet/model.py
@@ -133,6 +133,7 @@
 
 
 def preprocess_input(x, **kwargs):
+    kwargs = {k: v for k, v in kwargs.items() if k in ['backend', 'layers', 'models', 'utils']}
     return _preprocess_input(x, mode='torch', **kwargs)
 
 

Original file line number	Diff line number	Diff line change
`@@ -133,6 +133,7 @@`
`133`	`133`
`134`	`134`
`135`	`135`	`def preprocess_input(x, **kwargs):`
	`136`	`+ kwargs = {k: v for k, v in kwargs.items() if k in ['backend', 'layers', 'models', 'utils']}`
`136`	`137`	`return _preprocess_input(x, mode='torch', **kwargs)`
`137`	`138`
`138`	`139`