本文整理汇总了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) 示例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)