MODEL_VERSION = 1 | |
def check_model_version(test): | |
if test != MODEL_VERSION: | |
raise Exception( | |
f"Saved model version ({test}) does not match the "\ | |
f"source code model version ({MODEL_VERSION}). "\ | |
"Please pull the latest code or model checkpoints.") | |