File tree Expand file tree Collapse file tree 2 files changed +7
-1
lines changed Expand file tree Collapse file tree 2 files changed +7
-1
lines changed Original file line number Diff line number Diff line change @@ -566,6 +566,12 @@ def load_checkpoint(self,
566
566
567
567
# Preporcess distributed parameters.
568
568
file_name = os .path .join (checkpoint_dir , '.lock' )
569
+ meta_file = os .path .join (checkpoint_dir , 'meta.json' )
570
+ if not os .path .exists (meta_file ):
571
+ logger .error ("Please make sure the checkpoint dir {} exists, and "
572
+ "parameters in that dir are validating." .format (
573
+ checkpoint_dir ))
574
+ exit ()
569
575
distributed = self .loss_type in ["dist_softmax" , "dist_arcface" ]
570
576
if load_for_train and self .trainer_id == 0 and distributed :
571
577
self .process_distributed_params (checkpoint_dir )
Original file line number Diff line number Diff line change 12
12
# See the License for the specific language governing permissions and
13
13
# limitations under the License.
14
14
""" PLSC version string """
15
- plsc_version = "0.0.0 "
15
+ plsc_version = "0.1.1 "
You can’t perform that action at this time.
0 commit comments