Small fixes and updates.

hypernet.yaml

@@ -95,7 +95,7 @@ spec:
     - name: hypertrainer_image
       value: 'docker.io/gooseai/basedformer'
     - name: hypertrainer_tag
-      value: '0bde514'
+      value: '3b75904'
 
   templates:
   - name: main
@@ -287,7 +287,7 @@ spec:
         - name: warmup
     container:
       image: "{{workflow.parameters.hypertrainer_image}}:{{workflow.parameters.hypertrainer_tag}}"
-      command: [ "/usr/bin/python3", "/usr/src/app/hypertrain.py" ]
+      command: [ "/usr/bin/python3", "/app/hypertrain.py" ]
       args: ["--run_name", "{{inputs.parameters.run_name}}",
              "--model", "{{inputs.parameters.model}}",
              "--dataset", "{{inputs.parameters.dataset}}",

hypertrain.py

@@ -376,7 +376,7 @@ train_loader = torch_data.DataLoader(train_dataset,
                                      batch_size=bs * gas,
                                      shuffle=train_config["shuffle"],
                                      num_workers=0)
-wandb.init(project=train_config["project-id"],
+wandb.init(project=train_config["project_id"],
            name=train_config["run_name"],
            config={**train_config, **model.config})