Error in error message - models/eval/visualization.py

[albusdemens]

Current code (

pytorch_GAN_zoo/models/eval/visualization.py

Line 68 in 382a968

if checkpointData is None:

):

    if checkpointData is None:
        raise FileNotFoundError(
            "Not checkpoint found for model " + name + " at directory " + dir)

Should be changed into something like

    if checkpointData is None:
        print(
            "Checkpoint not found for model %s at directory %s" % (name, checkPointDir))