333 Commits

Author SHA1 Message Date
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