当前位置: 首页>>代码示例>>Python>>正文


Python model.CNN属性代码示例

本文整理汇总了Python中model.CNN属性的典型用法代码示例。如果您正苦于以下问题:Python model.CNN属性的具体用法?Python model.CNN怎么用?Python model.CNN使用的例子?那么, 这里精选的属性代码示例或许可以为您提供帮助。您也可以进一步了解该属性所在model的用法示例。


在下文中一共展示了model.CNN属性的2个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。

示例1: setup

# 需要导入模块: import model [as 别名]
# 或者: from model import CNN [as 别名]
def setup():
    global model, vocab, max_length
    # load vocab, vocab_size, max_length
    with open(os.path.join(PATH, 'vocab.json'), 'r') as fp:
        vocab = json.load(fp)

    with open(os.path.join(PATH, 'config.txt'), 'r') as f:
        vocab_size = int(re.sub('\n', '', f.readline()))
        max_length = int(f.readline())

    config = tf.ConfigProto()
    config.gpu_options.allow_growth = True
    sess = tf.Session(config=config)
    model = CNN(sess=sess, vocab_size=vocab_size, sequence_length=max_length, trainable=True)
    saver = tf.train.Saver()
    saver.restore(sess, tf.train.latest_checkpoint(PATH + '/models')) 
开发者ID:bzantium,项目名称:chatbot-seminar,代码行数:18,代码来源:common.py

示例2: model_creator

# 需要导入模块: import model [as 别名]
# 或者: from model import CNN [as 别名]
def model_creator(layers):
        model = CNN(32, 3, args.channels, 10, layers, n_nodes=args.nodes)
        criterion = nn.CrossEntropyLoss()

        optim = torch.optim.SGD(model.parameters(), 0.025, momentum=0.9, weight_decay=3.0E-4)
        lr_scheduler = torch.optim.lr_scheduler.CosineAnnealingLR(optim, args.epochs, eta_min=0.001)

        return model, criterion, optim, lr_scheduler 
开发者ID:microsoft,项目名称:nni,代码行数:10,代码来源:search.py


注:本文中的model.CNN属性示例由纯净天空整理自Github/MSDocs等开源代码及文档管理平台,相关代码片段筛选自各路编程大神贡献的开源项目,源码版权归原作者所有,传播和使用请参考对应项目的License;未经允许,请勿转载。