dvc.lock

schema: '2.0'
stages:
  data_ingestion:
    cmd: python src/cnnClassifier/pipeline/stage_01_data_ingestion.py
    deps:
    - path: config/config.yaml
      hash: md5
      md5: 1862bf8f61785025c134e54fb00dba89
      size: 802
    - path: src/cnnClassifier/pipeline/stage_01_data_ingestion.py
      hash: md5
      md5: dfc22d108828ff9f4b2b91880e939b76
      size: 913
    outs:
    - path: artifacts/data_ingestion/Eye_diseases
      hash: md5
      md5: eba44f61d3415210b04ad4dbe705939b.dir
      size: 2553843
      nfiles: 383
  prepare_base_model:
    cmd: python src/cnnClassifier/pipeline/stage_02_prepare_base_model.py
    deps:
    - path: config/config.yaml
      hash: md5
      md5: 1862bf8f61785025c134e54fb00dba89
      size: 802
    - path: src/cnnClassifier/pipeline/stage_02_prepare_base_model.py
      hash: md5
      md5: 263e76b3cce7b23e32cb2e68652cc632
      size: 1009
    params:
      params.yaml:
        CLASSES: 5
        IMAGE_SIZE:
        - 224
        - 224
        - 3
        INCLUDE_TOP: false
        LEARNING_RATE: 0.01
        WEIGHTS: imagenet
    outs:
    - path: artifacts/prepare_base_model
      hash: md5
      md5: 617a29715d6e1b304dbdb80d94d36b4e.dir
      size: 118360304
      nfiles: 2