Skip to content

Clean, reproducible, boilerplate-free deep learning project template.

Notifications You must be signed in to change notification settings

groomata/vision

Repository files navigation

그루마타 온라인 클래스에서 진행한 프로젝트입니다.

프로젝트는 pytorch-lightning을 사용하여 구성되었으며, 모든 실험 설정과 결과물은 wandb에 자동으로 기록됩니다. wandb workspace의 예시는 여기에서 확인할 수 있습니다.

실험 설정은 hydra에 의해 관리되기에 CLI를 통해 제어할 수 있습니다. 설정은 hydra-zen을 통해 자동으로, programmatic하게 생성됩니다.

Usage example

Trainingdocker를 이용하는 것이 권장됩니다. groomata/vision 이미지를 이용하면 되며, 상기했듯 모든 실험 설정은 CLI를 이용해 제어할 수 있습니다.

docker run \
    --gpus=all \
    --ipc=host \
    --volume=/path/to/volume:/vision/.cache \
    --env-file=/path/to/.env \
    --tty \
    groomata/vision \
    # 원하는 실험 설정으로 덮어 쓰면 됩니다.
    optimizer.lr=0.0001 \
    datamodule.dataloader.batch_size=64 \
    trainer.max_epochs=100 \
    trainer.gradient_clip_algorithm="norm" \
    trainer.gradient_clip_val=1.0

Note

프로젝트를 작성해나가는 과정은 그루마타 온라인 클래스에서 end-to-end로 경험할 수 있어요.