Skip to content

Commit

Permalink
removed stale script string
Browse files Browse the repository at this point in the history
  • Loading branch information
afoix committed Jul 21, 2024
1 parent 6e9ffcf commit ffce0d3
Showing 1 changed file with 0 additions and 9 deletions.
9 changes: 0 additions & 9 deletions scripts/shapeembed/slurm_sweep_shapeembed.py
Original file line number Diff line number Diff line change
Expand Up @@ -122,15 +122,6 @@ def find_submitted_slurm_jobs():
shapeembed_script=f'{os.getcwd()}/shapeembed.py'
wandb_project='shapeembed'

slurm_script="""#! /bin/bash
echo "running shape embed with:"
echo " - dataset {dataset[0]} ({dataset[1]}, {dataset[2]})"
echo " - model {model} ({model_params})"
echo " - compression_factor {compression_factor}"
echo " - batch size {batch_size}"
python3 shapeembed.py --wandb-project {wandb_project} --dataset {dataset[0]} {dataset[1]} {dataset[2]} --model {model} --compression-factor {compression_factor} --batch-size {batch_size} --clear-checkpoints --output-dir {out_dir} {extra_args}
"""

################################################################################

def spawn_slurm_job(slurm_out_dir, out_dir, ps, logger=logging.getLogger(__name__)):
Expand Down

0 comments on commit ffce0d3

Please sign in to comment.