Updates

brain2vec.py

@@ -576,7 +576,7 @@ def main():
     """
     Main function to parse command-line arguments and execute training or inference.
     """
-    parser = argparse.ArgumentParser(description="BRLP Lite Training and Inference Script")
+    parser = argparse.ArgumentParser(description="brain2vec Training and Inference Script")
 
     subparsers = parser.add_subparsers(dest='command', required=True, help='Sub-commands: train or infer')
 
@@ -595,7 +595,7 @@ def main():
     train_parser.add_argument('--aug_p', type=float, default=0.8, help='Augmentation probability.')
 
     # Inference Subparser
-    infer_parser = subparsers.add_parser('infererence', help='Run inference to encode images.')
+    infer_parser = subparsers.add_parser('inference', help='Run inference to encode images.')
     infer_parser.add_argument('--dataset_csv', type=str, required=True, help='Path to the dataset CSV file.')
     infer_parser.add_argument('--aekl_ckpt', type=str, required=True, help='Path to the autoencoder checkpoint.')
     infer_parser.add_argument('--output_dir', type=str, required=True, help='Directory to save latent representations.')
@@ -616,7 +616,7 @@ def main():
             lr=args.lr,
             aug_p=args.aug_p,
         )
-    elif args.command == 'infer':
+    elif args.command == 'inference':
         inference(
             dataset_csv=args.dataset_csv,
             aekl_ckpt=args.aekl_ckpt,