          self.name = "VGG Obstructed"
          self.input_size = 500
          self.vram = 3936
          self.vram_warnings = 1088  # at BS 1. OOMs at higher batchsizes
          self.vram_per_batch = 208
          self.batchsize = self.config["batch-size"]
  
      def init_model(self):
          self.model = KSession(self.name, self.model_path, model_kwargs=dict())
          self.model.load_model()
 
</s> remove         self.vram_per_batch = 96
 </s> add         self.vram_per_batch = 400 </s> remove         self.vram = 3104
 </s> add         self.vram = 2944 </s> remove         self.vram_per_batch = 48
 </s> add         self.vram_per_batch = 80 </s> remove         self.vram_per_batch = 128
 </s> add         self.vram_per_batch = 208 </s> remove         self.vram = 4096
 </s> add         self.vram = 4112
        self.vram_per_batch = 304
          self.name = "VGG Obstructed"
          self.input_size = 500
          self.vram = 3936
          self.vram_warnings = 1088  # at BS 1. OOMs at higher batchsizes
        self.vram_per_batch = 304
          self.batchsize = self.config["batch-size"]
  
      def init_model(self):
          self.model = KSession(self.name, self.model_path, model_kwargs=dict())
          self.model.load_model()
keep keep keep keep replace keep keep keep keep keep
 Optimize Extract parallel VRAM allocation 
plugins/extract/mask/vgg_obstructed.py