sweep.yaml

name: Pedram-sweep-Dev
program: prog_caller.py
method: bayes
metric: 
  goal: minimize
  name: MSE loss
parameters: 
  encoder: 
    values: [resnet18, resnet50, efficientb0, efficientb2, efficientb6]
  lr-gen: 
    min: 0.00001 
    max: 0.001
  lr-dis: 
    min: 0.000001
    max: 0.0001
  batch-size: 
    values: [64, 128]
  clip: 
    values: [1 , 2, 3, 4, 5, 6, 7, 8, 9]
  activation: 
    values: [ReLU, LeakyReLU]
  L2: 
    values: [True, False]

command:
  - ${env}
  - python3
  - ${program}
  - "--proc"
  - "TRAIN"
  - --num-epoch 
  - 5
  - --training-df-path 
  - /home/aghazap/Desktop/SimulationExecutables/SimPilotExe_V10_3_0_B140_Highway3LanesOneway/processed/processed_PA_M11_SpeedMatch_2023_02_21-13_45.pkl
  - --training-image-path 
  - /home/aghazap/Desktop/SimulationExecutables/SimPilotExe_V10_3_0_B140_Highway3LanesOneway/processed/images
  - --algo 
  - GAN
  - --track
  - --sweep
  - ${args}