check all conditions all the time; V0 -> V1 and V1 -> V2 do not suffice.
namespace caffe {
bool NetNeedsUpgrade(const NetParameter& net_param) {
- return NetNeedsV0ToV1Upgrade(net_param) || NetNeedsV1ToV2Upgrade(net_param);
+ return NetNeedsV0ToV1Upgrade(net_param) || NetNeedsV1ToV2Upgrade(net_param)
+ || NetNeedsDataUpgrade(net_param) || NetNeedsInputUpgrade(net_param);
}
bool UpgradeNetAsNeeded(const string& param_file, NetParameter* param) {
return 2;
}
bool need_upgrade = NetNeedsUpgrade(net_param);
- bool need_data_upgrade = NetNeedsDataUpgrade(net_param);
bool success = true;
if (need_upgrade) {
success = UpgradeNetAsNeeded(input_filename, &net_param);
LOG(ERROR) << "File already in latest proto format: " << input_filename;
}
- if (need_data_upgrade) {
- UpgradeNetDataTransformation(&net_param);
- }
-
// Save new format prototxt.
WriteProtoToTextFile(net_param, argv[2]);