fix compile for eval_cls.py

vitookit/evaluation/eval_cls.py

@@ -335,7 +335,7 @@ def main(args):
     if args.pretrained_weights:
         load_pretrained_weights(model, args.pretrained_weights, checkpoint_key=args.checkpoint_key, prefix=args.prefix)
     if args.compile:
-        return torch.compile(model)    
+        model = torch.compile(model)    
     trunc_normal_(model.head.weight, std=2e-5)
     
     model.to(device)