Open bestlyh opened 5 years ago
实际上train_model.py中的train函数仅仅是提供了一个通用的对三种网络的训练流程,其中的net_factory参数代表三种不同的网络,会在train函数被调用时由调用者给出Pnet、Rnet或者ONet的网络实例。 具体训练时是在train.py这个文件中实现的训练,这个文件中导入了train_model中的train函数,train_model.py只是提供了一个类似于工具类的部分。
实际上train_model.py中的train函数仅仅是提供了一个通用的对三种网络的训练流程,其中的net_factory参数代表三种不同的网络,会在train函数被调用时由调用者给出Pnet、Rnet或者ONet的网络实例。
具体训练时是在train.py这个文件中实现的训练,这个文件中导入了train_model中的train函数,train_model.py只是提供了一个类似于工具类的部分。