main.py

import subprocess
from subprocess import Popen
import os,shlex

GPU_NUMBER                        = 0
NUM_RUNS                          = 6
SWEEP_ID                          = '3rz09brd'
WANDB_USERNAME                    = 'dsingh93'
WANDB_KEY                         = '558a28f84276e6957673c5ad3656ea9b82cbd29d' 
SEED                              = -1
WB_ENTITY_NAME                    = 'uitmlg'
WB_PROJECT_NAME                   = 'ssl'

if os.environ.get('SINGULARITY_NAME'):
  WANDB_EXEC_LOC= os.path.join(os.sep,'users',os.path.expanduser('~'),'.local','bin','wandb')
else:
  WANDB_EXEC_LOC='wandb'


cmd_sweep ="""{} agent uitmlg/ssl/{}""".format(WANDB_EXEC_LOC,SWEEP_ID)
#cmd_sweep ="""python build_dataset.py --seed 42 --dataset mnist --nlabels 1000""".format(SWEEP_ID)

os.environ['WANDB_API_KEY']       = WANDB_KEY
os.environ['WANDB_USERNAME']      = WANDB_USERNAME
os.environ['GPU']                 = str(GPU_NUMBER)
os.environ['SEED']                = str(SEED)
os.environ['WB_ENTITY_NAME']      = str(WB_ENTITY_NAME)
os.environ['WB_PROJECT_NAME']     = str(WB_PROJECT_NAME)


cmd = cmd_sweep
procs = [ Popen(shlex.split(cmd),shell=False) for i in range(NUM_RUNS)]
for p in procs:
  p.wait()