[refactor] duplicate behaviour of init_checkpoint and ckpt_dir for resuming a training

[fcdl94]

There are two different parameters for doing the same stuff.

In particular:

• init_checkpoint does not change the actual checkpoint directory (same as output folder)
• ckpt_dir uses the value as the checkpoint directory (separated from the output folder)