dvc.yaml

stages:

  train:
    foreach: ${autoencoders}
    do:
      cmd: >
        python src/autoencoder/train.py
        --encoder-type ${item.type}
        --decoder-type ${item.type}
        --ae-type ${item.model}
        --model-path ${item.model_path}
        --epochs ${epochs}
        --batch-size ${batch_size}
        --latent-size ${latent_size}
        --log-dir ${item.log_dir}
        --lr ${lr}
        --val-ratio ${val_ratio}
        --reg-rate ${item.reg_rate}
        --reg-type ${item.reg_type}
        --gen-gifs true
      deps:
        - ${pokemon_sprites.data_dir}
        - src/autoencoder/train.py
        - ${item.model_src}
        - ${item.encoder_src}
        - ${item.decoder_src}
        - src/data/sprites.py
        - src/utils.py
      outs:
        - ${item.model_path}.pt
        - ${item.model_path}_kwargs.yaml
        - ${item.log_dir}/gen
        - ${item.log_dir}/latent
        - ${item.log_dir}/results
        - ${item.log_dir}/summary.txt:
            cache: false
      plots:
        - ${item.log_dir}/logs
      metrics:
        - ${item.log_dir}/logs.json:
            cache: false

  #create_serving_model:
    #foreach: ${autoencoders}
    #do:
      #cmd: >
        #torch-model-archiver 
        #--model-file ${item.model_file}
        #--model-name ${item.model}_${item.type} 
        #--serialized-file ${item.model_path}.pt
        #-v 1.0 
        #--handler image_classifier 
        #--export-path models/serving
      #deps:
        #- ${item.model_path}.pt
        #- ${item.model_path}_kwargs.yaml
      #outs:
        #- models/serving/${item.model}_${item.type}.mar