未验证 提交 4810142b 编写于 作者: L lilong12 提交者: GitHub

fix the hange issue if checkpoint_dir is specified but not exists and update version to 0.1.1 (#27)

* modify error message

* modify version to 0.1.1
上级 338eb998
......@@ -566,6 +566,12 @@ class Entry(object):
# Preporcess distributed parameters.
file_name = os.path.join(checkpoint_dir, '.lock')
meta_file = os.path.join(checkpoint_dir, 'meta.json')
if not os.path.exists(meta_file):
logger.error("Please make sure the checkpoint dir {} exists, and "
"parameters in that dir are validating.".format(
checkpoint_dir))
exit()
distributed = self.loss_type in ["dist_softmax", "dist_arcface"]
if load_for_train and self.trainer_id == 0 and distributed:
self.process_distributed_params(checkpoint_dir)
......
......@@ -12,4 +12,4 @@
# See the License for the specific language governing permissions and
# limitations under the License.
""" PLSC version string """
plsc_version = "0.0.0"
plsc_version = "0.1.1"
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册