training problem
Running trainer.
Choose one of saved models, or enter a name to create a new model. [r] : rename [d] : delete
[0] : new - latest : 0 Loading new_Quick96 model...
Choose one or several GPU idxs (separated by comma).
[CPU] : CPU [0] : NVIDIA GeForce RTX 3060
[0] Which GPU indexes to choose? : 0
Initializing models: 100%|###############################################################| 5/5 [00:01<00:00, 4.34it/s] Loading samples: 100%|###################################################################| 1/1 [00:00<00:00, 11.00it/s] Loading samples: 100%|###################################################################| 1/1 [00:00<00:00, 9.64it/s] ================ Model Summary ================= == == == Model name: new_Quick96 == == == == Current iteration: 2153 == == == ==-------------- Model Options ---------------== == == == batch_size: 4 == == == ==---------------- Running On ----------------== == == == Device index: 0 == == Name: NVIDIA GeForce RTX 3060 == == VRAM: 10.55GB == == ==
Starting. Press "Enter" to stop training and save model. Press any key to continue . . .
I also got the same situation. But when I tried again and again, suddenly it worked. This is not cool, but anyway I can train.
Experiencing same
Did you ever find the answer? If so, would you mind sharing it and closing this issue?