search

Daraan / CropAndWeedDetection

Using YOLOv7 for crop and weed detection
https://tensorboard.dev/experiment/wSv2959hQkCRQ34EAY4KzQ/#
MIT License
7 stars 3 forks source link

1. Can you support pytorch<2.0 for training? Computers do not support cuda for 11.8 #2

Open Kidgod77 opened 1 year ago

Kidgod77 commented 1 year ago
  1. Can you support pytorch<2.0 for training? Computers do not support cuda for 11.8 2 What are the differences between train experiment files and train? Is there a problem with the train file code? train experiment file I can use cpu to train, train will report error.
Daraan commented 1 year ago

I'll have a cleaning of the files & readme planned. Was quick release. Not sure if I get to it within this month.

1) PyTorch < 2.0 is in general possible and the first versions worked on it. Currently cannot estimate what needs to be changed to make the versions compatible again and what might break. I mostly worked with cuda 11.7.

2) The train.py file is set up to work with a single settings.json file and was build more with DeepSpeed in mind, the train_experiments.py to works with several input files from the experiments folder and is more recent. I'm planning to unify these.

train.py file has been fixed thank you. The --logname parameter was not optional like intended.