當前位置: 首頁>>代碼示例>>Python>>正文


Python tensorboardX.FileWriter方法代碼示例

本文整理匯總了Python中tensorboardX.FileWriter方法的典型用法代碼示例。如果您正苦於以下問題:Python tensorboardX.FileWriter方法的具體用法?Python tensorboardX.FileWriter怎麽用?Python tensorboardX.FileWriter使用的例子?那麽, 這裏精選的方法代碼示例或許可以為您提供幫助。您也可以進一步了解該方法所在tensorboardX的用法示例。


在下文中一共展示了tensorboardX.FileWriter方法的3個代碼示例,這些例子默認根據受歡迎程度排序。您可以為喜歡或者感覺有用的代碼點讚,您的評價將有助於係統推薦出更棒的Python代碼示例。

示例1: __init__

# 需要導入模塊: import tensorboardX [as 別名]
# 或者: from tensorboardX import FileWriter [as 別名]
def __init__(self, output_dir, data_loader, imsize):
        if cfg.TRAIN.FLAG:
            self.model_dir = os.path.join(output_dir, 'Model')
            self.image_dir = os.path.join(output_dir, 'Image')
            self.log_dir = os.path.join(output_dir, 'Log')
            mkdir_p(self.model_dir)
            mkdir_p(self.image_dir)
            mkdir_p(self.log_dir)
            self.summary_writer = FileWriter(self.log_dir)

        s_gpus = cfg.GPU_ID.split(',')
        self.gpus = [int(ix) for ix in s_gpus]
        self.num_gpus = len(self.gpus)
        torch.cuda.set_device(self.gpus[0])
        cudnn.benchmark = True

        self.batch_size = cfg.TRAIN.BATCH_SIZE * self.num_gpus
        self.max_epoch = cfg.TRAIN.MAX_EPOCH
        self.snapshot_interval = cfg.TRAIN.SNAPSHOT_INTERVAL

        self.data_loader = data_loader
        self.num_batches = len(self.data_loader) 
開發者ID:kkanshul,項目名稱:finegan,代碼行數:24,代碼來源:trainer.py

示例2: __init__

# 需要導入模塊: import tensorboardX [as 別名]
# 或者: from tensorboardX import FileWriter [as 別名]
def __init__(self, output_dir, data_loader, imsize):
        self.model_dir = os.path.join(output_dir, 'Model')
        self.image_dir = os.path.join(output_dir, 'Image')
        self.log_dir = os.path.join(output_dir, 'Log')
        mkdir_p(self.model_dir)
        mkdir_p(self.image_dir)
        mkdir_p(self.log_dir)
        self.summary_writer = FileWriter(self.log_dir)

        s_gpus = cfg.GPU_ID.split(',')
        self.gpus = [int(ix) for ix in s_gpus]
        self.num_gpus = len(self.gpus)
        torch.cuda.set_device(self.gpus[0])
        cudnn.benchmark = True

        self.batch_size = cfg.TRAIN.BATCH_SIZE * self.num_gpus
        self.max_epoch = cfg.TRAIN.MAX_EPOCH
        self.snapshot_interval = cfg.TRAIN.SNAPSHOT_INTERVAL

        self.data_loader = data_loader
        self.num_batches = len(self.data_loader) 
開發者ID:netanelyo,項目名稱:Recipe2ImageGAN,代碼行數:23,代碼來源:eval_trainer.py

示例3: __init__

# 需要導入模塊: import tensorboardX [as 別名]
# 或者: from tensorboardX import FileWriter [as 別名]
def __init__(self, model_dir):
        self.summary_writer = tb.FileWriter(model_dir) 
開發者ID:ramakanth-pasunuru,項目名稱:video_captioning_rl,代碼行數:4,代碼來源:tensorboard.py


注:本文中的tensorboardX.FileWriter方法示例由純淨天空整理自Github/MSDocs等開源代碼及文檔管理平台,相關代碼片段篩選自各路編程大神貢獻的開源項目,源碼版權歸原作者所有,傳播和使用請參考對應項目的License;未經允許,請勿轉載。