From 3f8e118fc841d3d2131a75ab56f4cdbfab0d6d51 Mon Sep 17 00:00:00 2001 From: gexy5 Date: Tue, 14 Jun 2022 15:35:03 +0800 Subject: [PATCH] add omnivore Signed-off-by: gexy5 --- omnivore.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/omnivore.py b/omnivore.py index 4759a75..ebebc72 100644 --- a/omnivore.py +++ b/omnivore.py @@ -101,7 +101,7 @@ class Omnivore(NNOperator): ) inputs = data.to(self.device)[None, ...] - feats = self.model.forward_features(inputs, input_type = self.input_type) + feats = self.model.forward_features(inputs) features = feats.to('cpu').squeeze(0).detach().numpy() outs = self.model.head(feats, input_type = self.input_type)