10 Commits

Author	SHA1	Message	Date
nielseni6	1ef6cbcec5	added pgt_coeff to argparser	2024-11-01 11:19:22 -04:00
nielseni6	21e660fde9	Fixed issues with pgt training, adding the pgt loss to the YOLO loss function. PGT loss can now be tracked during training.	2024-11-01 11:04:12 -04:00
nielseni6	9524af3bfe	Model loading using YOLOv10PGT added, and pgt_coeff is now a cfg parameter	2024-10-23 20:28:51 -04:00
nielseni6	943a333fae	Merge branch 'PGT_and_attribution' of https://github.com/nielseni6/yolov10 into PGT_and_attribution	2024-10-22 23:41:53 -04:00
nielseni6	0efbfe7f4d	updated grad to create_graph during training, enabling better pgt_loss optimization	2024-10-22 23:40:46 -04:00
nielseni6	2ccec65edc	fixed val and saving weights	2024-10-21 13:56:29 -04:00
nielseni6	411157c18a	fixed yaml default	2024-10-21 13:05:25 -04:00
nielseni6	5953d3c9c6	changed save path to not override previous training	2024-10-21 13:04:57 -04:00
nielseni6	3a449d5a6c	Fixed PGT by including it in the loss function	2024-10-21 12:20:24 -04:00
nielseni6	38fa59edf2	PGT Training now functioning. Run run_pgt_train.py to train the model.	2024-10-16 19:26:41 -04:00