              self.vram_free = free_mem[self.device]
  
          if self.verbose:
              print("GPU VRAM free:    {}".format(self.vram_free))
  
      def output_stats(self):
          """ Output stats in verbose mode """
 
</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 class GPUMem(object):
 </s> add class GPUMem(): </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
            if self.device == -1:
                print("No GPU. Limiting RAM useage to "
                      "{}MB".format(self.vram_free)) 
              self.vram_free = free_mem[self.device]
  
          if self.verbose:
            if self.device == -1:
                print("No GPU. Limiting RAM useage to "
                      "{}MB".format(self.vram_free)) 
              print("GPU VRAM free:    {}".format(self.vram_free))
  
      def output_stats(self):
          """ Output stats in verbose mode """
keep keep add keep keep keep keep
 Extract: Limit non-gpu RAM useage to 2GB 
lib/face_alignment/vram_allocation.py