Parse phase from flags
| 110 | |
| 111 | // Parse phase from flags |
| 112 | caffe::Phase get_phase_from_flags(caffe::Phase default_value) { |
| 113 | if (FLAGS_phase == "") |
| 114 | return default_value; |
| 115 | if (FLAGS_phase == "TRAIN") |
| 116 | return caffe::TRAIN; |
| 117 | if (FLAGS_phase == "TEST") |
| 118 | return caffe::TEST; |
| 119 | LOG(FATAL) << "phase must be \"TRAIN\" or \"TEST\""; |
| 120 | return caffe::TRAIN; // Avoid warning |
| 121 | } |
| 122 | |
| 123 | // Parse stages from flags |
| 124 | vector<string> get_stages_from_flags() { |