39 Commits (cad4d1618fee74471d335314cb77070fee951a42)

Author SHA1 Message Date
AlexeyAB cad4d1618f Added support for Tensor Cores CC >= 7.0 (V100). For FP16/32 (mixed precision) define CUDNN_HALF should be used. 7 years ago
AlexeyAB 033e934ce8 If there is excessive GPU-RAM consumption by CUDNN then then do not use Workspace 7 years ago
Joseph Redmon 352ae7e65b ADAM 9 years ago
Joseph Redmon 5c067dc447 good chance I didn't break anything 9 years ago
Joseph Redmon 8322a58cf6 hate warnings 9 years ago
Joseph Redmon c7b10ceadb so much need to commit 9 years ago
Joseph Redmon cff59ba135 go updates 9 years ago
Joseph Redmon 913d355ec1 lots of stuff 9 years ago
Joseph Redmon 9802287b58 some fixes 10 years ago
Joseph Redmon db0397cfaa shortcut layers, msr networks 10 years ago
Joseph Redmon 8fd18add6e CVPR Experiments 10 years ago
Joseph Redmon 846b3b4366 Added COCO, fixed memory leaks 10 years ago
Joseph Redmon a08ef29e08 NIGHTMARE!!!! 10 years ago
Joseph Redmon 516f019ba6 route handles input images well....ish 10 years ago
Joseph Redmon 664c5dd2f2 Subdivisions for batches 10 years ago
Joseph Redmon dcb000b553 refactoring and added DARK ZONE 10 years ago
Joseph Redmon 0f645836f1 Detection is back, baby\! 10 years ago
Joseph Redmon 979d02126b Generalizing conv layer so deconv is easier 10 years ago
Joseph Redmon 809f924db2 CUDA so fast 10 years ago
Joseph Redmon aa5996d58e Faster 11 years ago
Joseph Redmon 28e2115272 Distributed training 11 years ago
Joseph Redmon 1edcf73a73 Detection good, split up col images 11 years ago
Joseph Redmon 14303717dc Fast, needs to be faster 11 years ago
Joseph Redmon 787d534560 Convolutional working on GPU 11 years ago
Joseph Redmon 76ee68f96d Trying some stuff w/ dropout 11 years ago
Joseph Redmon d9f1b0b16e probably how maxpool layers should be 11 years ago
Joseph Redmon 70d622ea54 Added batch to col2im, padding option 11 years ago
Joseph Redmon cd8d53df21 So there WAS this huge bug. Gone now 11 years ago
Joseph Redmon 00d483697a Small updates 11 years ago
Joseph Redmon cc06817efa Attempt at visualizing ImageNet Features 11 years ago
Joseph Redmon 2ea63c0e99 Better VOC handling and resizing 11 years ago
Joseph Redmon 228d3663f8 Extracting features from VOC with temp filters 11 years ago
Joseph Redmon f7a17f82eb Convolutional layers working w/ matrices 11 years ago
Joseph Redmon b2b7137b6f About to do something stupid... 11 years ago
Joseph Redmon 0d6bb5d44d Working? 12 years ago
Joseph Redmon 2db9fbef2b Parsing, image loading, lots of stuff 12 years ago
Joseph Redmon d7286c2732 Loading may or may not work. But probably. 12 years ago
Joseph Redmon 9b1774bd39 Connected layers work forward and backward! 12 years ago
Joseph Redmon 41bcfac86f First Commit! 12 years ago