Small changes

harmonbhasin · Mar 31, 2024 · 94609e4 · 94609e4
1 parent 7f6a8ef
commit 94609e4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.gitignore b/.gitignore
@@ -1,4 +1,4 @@
 .ipynb_checkpoints
 __pycache__
-models.zip
 models
+.idea
diff --git a/src/train.py b/src/train.py
@@ -339,7 +339,7 @@ def main(args):
         with open(os.path.join(out_dir, "config.yaml"), "w") as yaml_file:
             yaml.dump(args.__dict__, yaml_file, default_flow_style=False)
 
-    api_key = 'a033029380789b03489a37e2e55903078f6f49be'
+    # Enter api key for wandb under variable api_key
     wandb.login(key=api_key)
 
     main(args)