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