Remove git artifact collection due to bugs

LukeBailey181 · Apr 25, 2023 · c21ec18 · c21ec18
1 parent ba32af4
commit c21ec18
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 6 deletions.
diff --git a/sweep_configs/cifar_standard_dropout_sweep.yaml b/sweep_configs/cifar_standard_dropout_sweep.yaml
@@ -1,4 +1,4 @@
-name: cifar_standard_dropout
+name: cifar_standard_dropout_282_seed
 description: Sweeping cifar p with standard dropout 
 project: k-dropout
 program: train_net.py
@@ -28,6 +28,8 @@ parameters:
   p:
     values: [0, 0.2, 0.5, 0.8]
   # dataset
+  seed:
+    value: 282
   dataset_name:
     value: cifar10
   batch_size:

diff --git a/train_net.py b/train_net.py
@@ -32,10 +32,10 @@
     if not args.local_only:
         # log the git diff and untracked files as an artifact
 
-        snapshot_name, snapshot_path = write_git_snapshot()
+        #snapshot_name, snapshot_path = write_git_snapshot()
 
         config = vars(args)
-        config["git_snapshot"] = snapshot_name
+        #config["git_snapshot"] = snapshot_name
         # TODO: specify run name in a more precise way
         #       e.g. with args.run_name_prefix and args.run_name_items which could
         #       include the params (p, k, etc..) to put in the run name
@@ -52,9 +52,10 @@
             run_name = args.run_name
         run = wandb.init(project="k-dropout", config=config, name=run_name)
 
-        snapshot_artifact = wandb.Artifact(snapshot_name, type="git_snapshot")
-        snapshot_artifact.add_file(snapshot_path)
-        wandb.log_artifact(snapshot_artifact)
+        # TODO fix artifact logging
+        #snapshot_artifact = wandb.Artifact(snapshot_name, type="git_snapshot")
+        #snapshot_artifact.add_file(snapshot_path)
+        #wandb.log_artifact(snapshot_artifact)
 
     # create model
     dropout_layer, layer_kwargs = get_dropout_layer(