用makefile作为神经网络训练/推理的入口

        makefile一般是用于编译一个软件系统,不过,你如果想要高效地进行神经网络的训练或者推理,不妨借助makefile来实现一个高效的入口。

       假如你有一个神经网络需要训练,假设名叫basenet,然后需要训练不同的参数下的版本,比较性能上的差异,于是可以在basenet中放置网络的定义,训练脚本train.py,推理脚本infer.py,还有其它需要的脚本...,然后建立和basenet平行的目录,例如basenet-1,basenet-2,...这些目录只需要放一个makefile以及训练参数的定义config.py,他们之间只有config.py有差异。

basenet目录

train.py定义一个函数入口train:


def train(end_epoch=10,batch_size=1,reload=0):
    print('training...',end_epoch, batch_size,reload)
    #do something
    print('done.')

infer.py定义了infer函数入口:


def infer(ckpt,infer_noisy_dir,saved_enhanced_dir):
    print('infer...',ckpt,infer_n
  • 2
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

geastwind1

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值