MCPcopy Index your code
hub / github.com/zai-org/CodeGeeX / check_exception_checkpoints

Function check_exception_checkpoints

codegeex/mindspore/train.py:408–422  ·  view source on GitHub ↗

Checkpoint exception checkpoints size. Args: ckpt_file_list: exception checkpoints Returns: result of exception checkpoints size check.

(ckpt_file_list)

Source from the content-addressed store, hash-verified

406
407
408def check_exception_checkpoints(ckpt_file_list):
409 """
410 Checkpoint exception checkpoints size.
411 Args:
412 ckpt_file_list: exception checkpoints
413 Returns: result of exception checkpoints size check.
414 """
415 ckpt_size_list = []
416 for ckpt_file in ckpt_file_list:
417 ckpt_size_list.append(os.path.getsize(ckpt_file))
418
419 if len(set(ckpt_size_list)) > 1:
420 return False
421
422 return True
423
424
425def restore_exception_checkpoint(args_param, sink_size, dataset, model, network, epoch):

Callers 1

Calls

no outgoing calls

Tested by

no test coverage detected