645 Commits

Author SHA1 Message Date
Glenn Jocher
a701374014 updates 2019-02-09 19:24:51 +01:00
Glenn Jocher
c2436d8197 updates 2019-02-08 22:43:05 +01:00
Glenn Jocher
8b88e50f2f updates 2019-02-08 16:50:48 +01:00
Glenn Jocher
eec0dc7b6c ONNX compatibility updates 2018-12-28 20:09:06 +01:00
Glenn Jocher
bf23be9965 updates 2018-12-16 15:16:52 +01:00
Glenn Jocher
18ccd184bf updates 2018-12-16 15:16:19 +01:00
Glenn Jocher
900851200e updates 2018-12-15 20:52:35 +01:00
Glenn Jocher
3c95b5c104 updates 2018-12-12 17:26:46 +01:00
Glenn Jocher
b5a2747a6a updates 2018-12-12 17:02:37 +01:00
Glenn Jocher
c591936446 updates 2018-12-11 21:49:56 +01:00
Glenn Jocher
3fe3951268 updates 2018-12-10 13:19:13 +01:00
Guillermo García
d03ce45da5 train.py freeze-darknet53 shortened to freeze and action store_true
Traing with freeze: python train.py --freeze
Train without freeze: python train.py

Note: in the actual version freeze is only for first epoche
2018-12-05 16:57:16 +01:00
Guillermo García
868a116750 train.py remove hardcoded weights/ path for weights.
If I want to store my weights in 'weights2' path:
python train.py --weights-path weights2

Default is the same: weights
2018-12-05 16:57:16 +01:00
Guillermo García
9c0c1f23ab scripts: use data config defined class names
Shorten name of --data-config-path argument to --data-config
2018-12-05 16:57:16 +01:00
Guillermo García
89daa407e5 train.py report argument as store_true
Default is false: python train.py
If want the report: python train.py --report
2018-12-05 16:57:16 +01:00
Guillermo García
b1fb6fa33d train.py resume argument as store_true
Default is false.

If want to resume, call train.py --resume
2018-12-05 16:57:16 +01:00
Guillermo García
c807c16b79 Fix argument parser bad practice
Keep parsing inside __main__ block and call methods with arguments

Add double -- for long argument names (- reserved for shortcuts)
2018-12-05 16:57:16 +01:00
Guillermo García
5a566454f5 Extract seed and cuda initialization utils 2018-12-05 11:55:27 +01:00
Glenn Jocher
fd6619d773 updates 2018-12-03 22:33:25 +01:00
Glenn Jocher
10cca39934 updates 2018-12-03 21:08:45 +01:00
Glenn Jocher
b64620cf75 updates 2018-12-03 14:12:46 +01:00
Glenn Jocher
40b536a426 updates 2018-12-03 14:08:59 +01:00
Glenn Jocher
5843c41dfc add multi_scale support 2018-12-03 14:05:50 +01:00
Glenn Jocher
af0033c9e9 updates 2018-11-29 11:59:29 +01:00
Glenn Jocher
053566b174 updates 2018-11-28 10:27:55 +01:00
Glenn Jocher
cc419d88ea updates 2018-11-28 10:25:00 +01:00
Glenn Jocher
5a0575af3a updates 2018-11-27 18:43:46 +01:00
Glenn Jocher
b07ee41867 updates 2018-11-27 18:14:48 +01:00
Glenn Jocher
ab9ee6aa9a updates 2018-11-23 19:45:39 +01:00
Glenn Jocher
887ab29c64 updates 2018-11-22 20:03:09 +01:00
Glenn Jocher
6f83c321c8 updates 2018-11-22 20:02:11 +01:00
Glenn Jocher
b9d87be318 updates 2018-11-22 14:54:52 +01:00
Glenn Jocher
db515a4535 updates 2018-11-22 14:14:19 +01:00
Glenn Jocher
809667404f updates 2018-11-22 13:52:22 +01:00
Glenn Jocher
4e4b67b3c5 updates 2018-11-21 18:01:18 +01:00
Glenn Jocher
a021f97110 updates 2018-11-15 01:01:04 +01:00
Glenn Jocher
1ea87c49c4 updates 2018-11-15 00:57:15 +01:00
Glenn Jocher
45c5567723 mAP recorded during training 2018-11-14 15:14:41 +00:00
Glenn Jocher
4bae1d0f75 updates 2018-11-09 17:03:26 +01:00
Glenn Jocher
b1a2735338 updates 2018-11-09 16:58:32 +01:00
Glenn Jocher
5177f3e7a0 updates 2018-11-09 16:48:55 +01:00
Glenn Jocher
664cbaab09 Adam optimizer 2018-11-09 16:44:12 +01:00
Glenn Jocher
46a4de77cb updates 2018-11-08 12:29:35 +01:00
Glenn Jocher
2463030d6c updates 2018-11-08 12:28:19 +01:00
Glenn Jocher
c8e4a19879 updates 2018-11-08 12:27:14 +01:00
Glenn Jocher
a6d69cefe0 updates 2018-11-08 12:26:23 +01:00
Glenn Jocher
edfad8095d updates 2018-11-05 23:34:26 +01:00
Glenn Jocher
6e5da1ce27 updates 2018-11-05 23:32:36 +01:00
Glenn Jocher
19ccb41eaf updates 2018-11-05 23:28:10 +01:00
Glenn Jocher
0096bb4dd5 update multi-scale training 2018-11-05 23:20:21 +01:00