Always set batchsize as int (#913)

deepfakes · Oct 26, 2019 · 9af7726 · 9af7726
1 parent 6efafab
commit 9af7726
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/plugins/extract/pipeline.py b/plugins/extract/pipeline.py
@@ -460,7 +460,7 @@ def _set_extractor_batchsize(self):
     @staticmethod
     def _set_plugin_batchsize(plugin, available_vram):
         """ Set the batch size for the given plugin based on given available vram """
-        plugin.batchsize = max(1, available_vram // plugin.vram_per_batch)
+        plugin.batchsize = int(max(1, available_vram // plugin.vram_per_batch))
         logger.verbose("Reset batchsize for %s to %s", plugin.name, plugin.batchsize)
 
     def _join_threads(self):