search

facebookresearch / pytorch_GAN_zoo

A mix of GAN implementations including progressive growing
BSD 3-Clause "New" or "Revised" License
1.61k stars 270 forks source link

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

Open albusdemens opened 4 years ago

albusdemens commented 4 years ago

Current code (https://github.com/facebookresearch/pytorch_GAN_zoo/blob/382a96816f65235d2467b4d13ca4c4ed587f874a/models/eval/visualization.py#L68):

    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))