chenyu
74c6cf8be3
lint mlperf model_train ( #10038 )
2025-04-24 16:19:44 -04:00
Francis Lata
defa1e77f6
get the proper dataset count ( #9962 )
2025-04-21 12:11:37 -04:00
Francis Lata
7bb36d71b2
remove openimages iterate ( #9820 )
2025-04-09 22:54:12 -04:00
Francis Lata
71b8890dd6
use validation dataloader inside retinanet eval ( #9747 )
2025-04-05 16:46:55 -04:00
Francis Lata
eb95825eea
RetinaNet dataloader ( #9442 )
...
* retinanet dataloader
* remove batch_size from generate_anchors
* refactor kits19 dataset tests
* add tests for dataloader
* fix testing setup and cleanups
* remove unused import
2025-03-21 13:36:41 -04:00
Francis Lata
c25d5d3101
improve isin checks ( #8589 )
2025-01-13 12:12:31 -05:00
Francis Lata
d3a387be63
[MLPerf] Prepare openimages dataset script ( #6747 )
...
* prepare openimages for MLPerf
* cleanup
* fix issue when clearing jit_cache on retinanet eval
* revert pandas specific changes
2024-09-27 11:13:56 -04:00
wozeparrot
a02b38c0ac
download openimages by running it ( #5396 )
2024-07-11 16:06:13 -07:00
Nik
085c0bbf6b
add mlperf train subset of openimages ( #4841 )
2024-06-05 10:10:11 -04:00
Quentin Wach
89b8b5d549
Fix missing import. ( #3666 )
2024-03-09 14:55:23 -08:00
George Hotz
d87a246439
move to new cached fetch ( #2493 )
...
* move to new cached fetch
* extra.utils is over
* loads
* bump download cache
* bump timeout
2023-11-28 17:36:55 -08:00
Jacky Lee
e0c2ae8984
Update file paths ( #1179 )
2023-07-07 18:41:58 -07:00
terafo
aa60feda48
Fix naming conflict with huggingface datasets ( #1161 )
...
* Rename in files
* Move files
* Moved to extra/datasets as suggested
* Changes to files
* Fixed stupid mistake
---------
Co-authored-by: terafo <terafo@protonmail.com >
2023-07-07 10:43:44 -07:00
