Fixed a bug in FLOP computation

predtuner/torchapp.py

@@ -229,7 +229,7 @@ class TorchApp(ModeledApp, abc.ABC):
         return module_indexer.module
 
     def _sample_input(self):
-        inputs, _ = next(iter(self.tune_loader))
+        inputs, _ = next(iter(DataLoader(self.tune_loader.dataset, batch_size=1)))
         return inputs.to(self.device)