卷积神经网络训练工具
convtt的Python项目详细描述
convtt
卷积神经网络训练工具
Densenet培训示例
该示例包含在文件名为convtt_train_densenet.py
的包中的bin
文件夹中。
下面是创建一个培训器并在给定模型和数据集的情况下训练网络的代码片段。
# convtt_train_densenet.pyfromconvtt.modelsimportdensenetfromconvtt.train.trainerimport*# initialise traineroptimiser=build_optimiser(model=model,name='ScheduledSGD',milestones=[10,20],lr=0.1)driver=build_driver(model=model,training_epoch=30,batch_size=128,training_data=dataset.train['images'],training_label=dataset.train['labels'],validation_data=None,validation_label=None,test_data=dataset.test['images'],test_label=dataset.test['labels'],optimiser=optimiser)trainer=build_trainer(optimiser=optimiser,driver=driver)test_acc=trainer.eval()print(test_acc)