Where is using checkpoint in C2W4_Assignment

Hey there,

Can any one tell me who and where is using checkpoint in the C2W4_Assignment, or I missed implicit usages?

thanks.

I think there should be a call right after TEST in the training loop

i.e checkpoint.save(checkpoint_prefix)

something like:

  # TEST LOOP
  for x in test_dist_dataset:
    distributed_test_step(x)
  if epoch % 3 == 0: checkpoint.save(checkpoint_prefix)