              return
          print("\n----- Initial GPU Stats -----")
          self.stats.print_info()
          print("GPU VRAM free:    {}".format(self.vram_free))
          print("-----------------------------\n")
  
 
</s> remove             # Limit ram usage to 3072 for CPU
            self.vram_free = 3072
        else:        
 </s> add             # Limit RAM to 2GB for non-gpu
            self.vram_free = 2048
        else: </s> remove             # Limit ram usage to 3072 for CPU
            self.vram_total = 3072
 </s> add             # Limit ram usage to 2048 for CPU
            self.vram_total = 2048 </s> remove class GPUMem(object):
 </s> add class GPUMem():
        if self.device == -1:
            print("No GPU. Limiting RAM useage to {}MB".format(self.vram_free))
              return
          print("\n----- Initial GPU Stats -----")
        if self.device == -1:
            print("No GPU. Limiting RAM useage to {}MB".format(self.vram_free))
          self.stats.print_info()
          print("GPU VRAM free:    {}".format(self.vram_free))
          print("-----------------------------\n")
  
keep add keep keep keep keep
 Extract: Limit non-gpu RAM useage to 2GB 
lib/face_alignment/vram_allocation.py