How to train a model without any labels?

csaroff / dinov2

PyTorch code and models for the DINOv2 self-supervised learning method.

Apache License 2.0

2 stars 2 forks source link

How to train a model without any labels? #1

Open PMRS-lab opened 2 months ago

PMRS-lab commented 2 months ago

Hello, Thank you for your earlier answer，My GPU device is RTX3090, I'm not sure if your code can easily run training on A100 nodes with the skypilot？

PMRS-lab commented 2 months ago

On the original version of Binov2, I ran the/run/train/train.exe file with the following code and found that it never started training or generated any error messages. python dinov2/run/train/train.py --nodes 1 --config-file dinov2/configs/train/vitl16_short.yaml --output-dir <PATH/TO/OUTPUT/DIR> train.dataset_path=ImageNet:split=TRAIN:root=<PATH/TO/DATASET>:extra=<PATH/TO/DATASET>