pyproject.toml

[tool.poetry]
name = "convolutional-neural-network"
version = "1.0.0"
description = ""
authors = ["Hongbo Miao"]

[tool.poetry.dependencies]
python = "3.10.x"
dvc = "2.38.1"
dvc-s3 = "2.21.0"
pandas = "1.5.2"
tabulate = "0.9.0"
torch = "1.13.1"
torchvision = "0.14.1"
wandb = "0.13.7"

[tool.poetry.group.dev.dependencies]
poethepoet = "0.16.5"

[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"

[tool.poe.tasks]
dev = "python main.py"
dev-download-original-data = "python main.py --should_download_original_data=True"

dvc-init = "dvc init --subdir"
dvc-add = "dvc add data/raw/cifar-10-python.tar.gz"
dvc-run-prepare = "dvc run --name=prepare --deps=data/raw/cifar-10-python.tar.gz --outs=data/processed/cifar-10-batches-py/ tar -x -f data/raw/cifar-10-python.tar.gz -C data/processed"
dvc-run-main = "dvc run --name=main --deps=data/processed/cifar-10-batches-py/ --deps=model/ --deps=main.py --deps=train.py --deps=evaluate.py --params=lr,train.epochs --outs=output/models/model.pt python main.py"
dvc-run-main-no-exec = "dvc run --no-exec --name=main --deps=data/processed/cifar-10-batches-py/ --deps=model/ --deps=main.py --deps=train.py --deps=evaluate.py --params=lr,train.epochs --outs=output/models/model.pt python main.py"
dvc-repro = "dvc repro data/raw/cifar-10-python.tar.gz.dvc"

dvc-remove-stage = "dvc remove STAGE_NAME"
dvc-dag = "dvc dag"

dvc-remote-add = "dvc remote add -d hm-remote s3://hongbomiao-bucket/hm-cnn-dvc-store"
dvc-push = "dvc push"
dvc-pull = "dvc pull"