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