+######################################################################
+
+nb_epochs_finished = 0
+
+try:
+ checkpoint = torch.load(args.checkpoint_name, map_location = device)
+ nb_epochs_finished = checkpoint['nb_epochs_finished']
+ model.load_state_dict(checkpoint['model_state'])
+ optimizer.load_state_dict(checkpoint['optimizer_state'])
+ print(f'Checkpoint loaded with {nb_epochs_finished} epochs finished.')
+
+except FileNotFoundError:
+ print('Starting from scratch.')
+
+except:
+ print('Error when loading the checkpoint.')
+ exit(1)
+
+######################################################################
+
+for k in range(nb_epochs_finished, args.nb_epochs):