首页 文章

在张量流中训练具有不同数据集的相同模型

提问于
浏览
0

问题:我有一个我想用独立数据集训练的模型 . 之后,我想提取每个模型的权重(每个实例的模型相同,但使用不同的数据集进行训练),最后计算这些权重的平均值 . 基本上,我的目的是模仿在多个设备上运行的张量流,然后平均它们的权重,以便它们被一个模型使用 .

我的解决方案:我将此模型多次添加到tensorflow,我目前正在使用其独特的数据集分别训练每个模型..但这是使用GB的内存,我想知道是否有更好的方法来做到这一点?

1 回答

  • 0

    其中一个可能的解决方案是,您可以使用其他类似的网络(类似的数据集,即,如果您的数据集是图像,您可以使用AlexNet权重)微调网络权重,请't afraid if your network has no same architecture, you can simply load weights of layers as much as you need by ' load_with_skip'函数https://github.com/joelthchao/tensorflow-finetune-flickr-style/blob/master/network.py

    微调比从头开始训练网络要少得多,

相关问题