Warn when layers are not loaded from checkpoint

[szymanowiczs]

Previously the script would not provide any indication when the model weights are not loaded from checkpoint, potentially leading to hard-to-find bugs.