Commit Graph

99 Commits (d3e77781517fe5a28bd929b947278c832f95c036)

Author SHA1 Message Date
Glenn Jocher 569fbc63c8 export.py update 2020-07-03 11:29:53 -07:00
Glenn Jocher a03adb5ab8 Merge remote-tracking branch 'origin/master' 2020-07-01 19:16:06 -07:00
Glenn Jocher 1fca7a7f24 autopad() update in common.py 2020-07-01 19:15:59 -07:00
Glenn Jocher a62a1c2c67 export.py update 2020-07-01 16:14:49 -07:00
Glenn Jocher f1d67f4110 update export.py 2020-07-01 15:46:15 -07:00
Glenn Jocher b5659d1195 module updates 2020-07-01 11:44:49 -07:00
Glenn Jocher 615d6d0cfa FROM nvcr.io/nvidia/pytorch:20.06-py3 2020-06-29 14:00:13 -07:00
Glenn Jocher e8bf385795
Merge pull request #223 from ArmyOfRobots/master
Adding torchscript export
2020-06-29 13:39:53 -07:00
Glenn Jocher 0bc80e1e57 onnx_export.py bug fix #224 2020-06-28 17:07:03 -07:00
Jake Poznanski 685091a3fc Adding torchscript export 2020-06-28 13:17:05 -07:00
Glenn Jocher 9a9c4f1259 update yolo.py default --cfg 2020-06-24 18:00:43 -07:00
Glenn Jocher e071b8dd3a update models/common.py for Conv() flexible padding 2020-06-24 13:12:23 -07:00
Glenn Jocher 0825cb7fd8 create models/hub 2020-06-24 11:58:23 -07:00
Glenn Jocher b1b3634145 update onnx_export.py #134 2020-06-24 11:00:03 -07:00
Glenn Jocher 8699c319ca update yolo.py 2020-06-24 10:42:01 -07:00
Glenn Jocher 53294f28b4 update yolo.py 2020-06-23 10:53:42 -07:00
Glenn Jocher 364fcfd7db PANet update 2020-06-22 14:27:17 -07:00
Glenn Jocher fc171e2678 check_anchor_order() update 2020-06-21 14:08:59 -07:00
Glenn Jocher e249792c2f update onnx_export.py usage 2020-06-19 10:15:08 -07:00
Glenn Jocher 80b82e8bde update yolo.py 2020-06-18 19:33:02 -07:00
Glenn Jocher e1e3399286 ONNX export bug fix #93 2020-06-16 09:59:42 -07:00
ashing 62eda5e9c2 add "__init__.py" file in <models> folder 2020-06-15 14:37:07 +08:00
Glenn Jocher 4ba47b1b1b model.yaml updates 2020-06-12 15:41:46 -07:00
Glenn Jocher 1e2cb6b480 ONNX export explicit cpu map_location 2020-06-12 13:54:00 -07:00
Glenn Jocher 8f17a96ff0 cleanup 2020-06-11 21:55:22 -07:00
Glenn Jocher 8f1755290c ONNX export update 2020-06-11 21:54:01 -07:00
Glenn Jocher 3a5c5328c5 ONNX, BCEBlurWithLogitsLoss, plot_study updates 2020-06-11 14:44:09 -07:00
Glenn Jocher a814720403 PyTorch Hub updates 2020-06-10 19:11:11 -07:00
Glenn Jocher c14368d768 new CSP model release 2020-06-09 14:31:04 -07:00
Glenn Jocher 854ccb9a74 _print_weights() 2020-06-09 10:51:47 -07:00
Glenn Jocher 5bee686649 model definition update 2020-06-08 23:26:56 -07:00
Glenn Jocher 715cb08b10 module updates 2020-06-08 22:15:16 -07:00
Glenn Jocher ff02ae0869 module updates 2020-06-08 22:13:01 -07:00
Glenn Jocher 12b0c046d5 model fusion and onnx export 2020-06-07 13:42:33 -07:00
Glenn Jocher b810b21299 augmented inference 2020-06-05 21:14:15 -07:00
Glenn Jocher 5c470d235e multi-gpu test bug fix #15 2020-06-05 20:36:29 -07:00
Glenn Jocher 67d5e65c77 profiling update 2020-06-05 20:22:39 -07:00
Glenn Jocher 2703ac73e8 profiling update 2020-06-05 20:22:17 -07:00
Glenn Jocher 4e32b6029a update 2020-06-05 20:03:21 -07:00
Glenn Jocher dbdee3a4a3 multi-gpu test bug fix #15 2020-06-04 15:23:03 -07:00
Glenn Jocher a2336088f0 updates 2020-06-04 14:25:10 -07:00
Glenn Jocher 17e4926cdf updates 2020-06-02 23:40:14 -07:00
Glenn Jocher ee8988b8a2 datasets.py self.batch_shapes fix 2020-06-01 18:11:18 -07:00
Glenn Jocher 7a00a5eff2 onnx_export.py 2020-06-01 14:05:01 -07:00
Glenn Jocher df7988d8d0 onnx_export.py 2020-06-01 13:53:47 -07:00
Glenn Jocher b10609fe6b updates 2020-05-31 15:07:04 -07:00
Glenn Jocher f129cdda9f updates 2020-05-30 11:03:50 -07:00
Glenn Jocher 1b97392801 updates 2020-05-30 10:46:35 -07:00
Glenn Jocher 1e84a23f38 initial commit 2020-05-29 17:04:54 -07:00