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

Python cfg.GPU_ID屬性代碼示例

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


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

示例1: __init__

# 需要導入模塊: from miscc.config import cfg [as 別名]
# 或者: from miscc.config.cfg import GPU_ID [as 別名]
def __init__(self, output_dir, data_loader, n_words, ixtoword, dataset):
        if cfg.TRAIN.FLAG:
            self.model_dir = os.path.join(output_dir, 'Model')
            self.image_dir = os.path.join(output_dir, 'Image')
            mkdir_p(self.model_dir)
            mkdir_p(self.image_dir)

        #torch.cuda.set_device(cfg.GPU_ID)
        #cudnn.benchmark = True

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

        self.n_words = n_words
        self.ixtoword = ixtoword
        self.data_loader = data_loader
        self.dataset = dataset
        self.num_batches = len(self.data_loader)
開發者ID:MinfengZhu，項目名稱:DM-GAN，代碼行數:21，代碼來源:trainer.py

示例2: __init__

# 需要導入模塊: from miscc.config import cfg [as 別名]
# 或者: from miscc.config.cfg import GPU_ID [as 別名]
def __init__(self, output_dir, data_loader, n_words, ixtoword):
        if cfg.TRAIN.FLAG:
            self.model_dir = os.path.join(output_dir, 'Model')
            self.image_dir = os.path.join(output_dir, 'Image')
            mkdir_p(self.model_dir)
            mkdir_p(self.image_dir)

        torch.cuda.set_device(cfg.GPU_ID)
        cudnn.benchmark = True

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

        self.n_words = n_words
        self.ixtoword = ixtoword
        self.data_loader = data_loader
        self.num_batches = len(self.data_loader)
開發者ID:Wentong-DST，項目名稱:attn-gan，代碼行數:20，代碼來源:trainer.py


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