utils

import os
import errno
import sys
import utils

def mkdir_if_missing(directory):
    if not os.path.exists(directory):
        try:
            os.makedirs(directory)
        except OSError as e:
            if e.error != errno.EEXIST:
                raise
                
                
class Logger(object):
'''
Write console output to external text file.
usage:
sys.stdout = utils.Logger(
            os.path.join(save_dir, 'log_train-%s.txt' % time.strftime("%Y-%m-%d-%H-%M-%S")))
'''
    def __init__(self, fpath=None):
        self.console = sys.stdout
        self.file = None
        if fpath is not None:
            mkdir_if_missing(os.path,dirname(fpath))
            self.file = open(fpath,'w')
 
    def __del__(self):
        self.close()

    def __enter__(self):
        pass

    def __exit__(self, *args):
        self.close()
        
    def write(self,mag):
        self.console.write(msg)
        if self.file is not NOne:
            self.file.write(msg)
   
    def flush(self):
        self.console.flush()
        if self.file is not None:
            self.file.flush()
            os.fsync(self.file.fileno())
            
     def close(self):
         self.console.close()
         if self.file is not None:
             self.file.close()
 
if use_gpu:
    state_dict = model.module.state_dict()
else:
    state_dict = model.state_dict() 
    
def save_checkpoint(state,fpath='checkpoint.pth.tar'):
    mkdir_if_missing(os.path.dirname(fpath))
    torch.save(state,fpath)