vitin1k

conda create -n vitin1k python=3.10 && conda activate vitin1k
pip install --index-url https://download.pytorch.org/whl/cu124 torch==2.4.1
pip install -r requirements.txt
conda install nvidia/label/cuda-12.4.0::cuda
python main.py {fit,validate,test,predict} -c config.yaml --root /path/to/folder/with/datasets

Name		Name	Last commit message	Last commit date
Latest commit History 5 Commits
datasets		datasets
models		models
training		training
.gitignore		.gitignore
README.md		README.md
classification.yaml		classification.yaml
main.py		main.py
requirements.txt		requirements.txt
segmentation.yaml		segmentation.yaml

Provide feedback