OWOD icon indicating copy to clipboard operation
OWOD copied to clipboard

If I only have one GPU, how should I set the training parameters?

Open lzylzylzy123456 opened this issue 1 year ago • 1 comments

As mentioned, I only have one RTX 4080, and I want to replicate the paper's results as closely as possible. I'm unsure about the optimal training parameter settings as I've noticed that training takes a considerable amount of time. Do you have any prior knowledge or recommendations? This would greatly help me save time!

lzylzylzy123456 avatar Sep 07 '23 07:09 lzylzylzy123456

python tools/train_net.py --num-gpus 1 --dist-url='tcp://127.0.0.1:52125' --resume --config-file ./configs/OWOD/t1/t1_train.yaml SOLVER.IMS_PER_BATCH 1 SOLVER.BASE_LR 0.0025 OUTPUT_DIR "./output/t1"

try it bro

SAistheBESTone avatar Mar 20 '24 12:03 SAistheBESTone