File size: 297 Bytes
b3c0032
c1a41d7
 
 
 
 
 
b3c0032
1
2
3
4
5
6
7
8
9
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.")