Davide Macario

@cooper-him I would also wrap the `del` statements in a try-except, as `checkpoint` may not be defined. In any case, I tested the code (training a model with the same...