391 Commits

Author SHA1 Message Date
Glenn Jocher
120d40c06a
Update train.py
This updates the PR to a one-liner to minimize additions. Perhaps we can include opt in the future but let's start with this for now.
2020-07-14 12:32:08 -07:00
Janko Ondras
38acc5f3c5 Fix img_size naming in hyperparameters logging 2020-07-14 08:48:46 +02:00
Janko Ondras
e558963244 Log hyperparameters in tensorboard
Log both hyperparameters and command line options in tensorboard.
2020-07-13 12:55:43 +02:00
Glenn Jocher
e169edfcf5
Update README.md 2020-07-12 14:28:19 -07:00
Glenn Jocher
8412d44dce update datasets.py to add detect.py --source glob_command feature 2020-07-12 14:22:15 -07:00
Glenn Jocher
eae33303d6 update datasets.py LoadImages() path improvements and Mixup 2020-07-12 14:14:51 -07:00
Glenn Jocher
0afbb8d498 update test.py model.half() 2020-07-12 13:25:58 -07:00
Glenn Jocher
41ff83a98d update torch_utils.py to FP16 EMA 2020-07-12 12:18:00 -07:00
Glenn Jocher
8b38e6f429 update dataset comments 2020-07-12 10:37:06 -07:00
Glenn Jocher
1b1681bac9 Created using Colaboratory 2020-07-11 15:25:52 -07:00
Glenn Jocher
9d7c7784f2 Created using Colaboratory 2020-07-11 15:16:24 -07:00
Glenn Jocher
a5818b280e expired pytorch hub cache bug fix #365 2020-07-11 15:04:53 -07:00
Glenn Jocher
8797246948 Merge remote-tracking branch 'origin/master' 2020-07-11 15:03:18 -07:00
Glenn Jocher
6e3c3b62b2 expired pytorch hub cache bug fix #365 2020-07-11 15:03:12 -07:00
Glenn Jocher
80d5775a57
Update README.md 2020-07-11 14:31:43 -07:00
Glenn Jocher
f7a6d6c8d4 remove https:// prefix from curl commands fix #355 2020-07-11 13:00:24 -07:00
Glenn Jocher
01a73ec08e multi-gpu ckpt filesize bug fix #253 2020-07-11 12:39:27 -07:00
Glenn Jocher
a586751904 multi-gpu ckpt filesize bug fix #253 2020-07-11 12:35:21 -07:00
Glenn Jocher
5de4e25d68 update tensorboard metric 2020-07-11 11:39:02 -07:00
Glenn Jocher
224e9a1d0b Merge remote-tracking branch 'origin/master' 2020-07-11 09:31:59 -07:00
Glenn Jocher
98fc483abc train.py results.txt to bucket bug fix 2020-07-11 09:31:53 -07:00
Glenn Jocher
d6f6b33a49 update datasets.py for image size check during caching 2020-07-11 00:21:52 -07:00
Glenn Jocher
01a67a9742 Merge remote-tracking branch 'origin/master' 2020-07-10 20:36:26 -07:00
Glenn Jocher
d3e786ed93 google_utils.py bug fix #355 2020-07-10 20:36:10 -07:00
Glenn Jocher
9006b85d21 update export.py torchscript file extension 2020-07-10 11:56:01 -07:00
Glenn Jocher
41bb70b325 update create_pretrained() 2020-07-10 00:25:35 -07:00
Glenn Jocher
1531795f5e avoid git status check in Docker bug fix #342 2020-07-10 00:11:50 -07:00
Glenn Jocher
1e9cf6a6e9 add file sizes to ckpt ops 2020-07-10 00:00:19 -07:00
Glenn Jocher
c494a11128 Merge remote-tracking branch 'origin/master'
# Conflicts:
#	train.py
2020-07-09 22:09:40 -07:00
Glenn Jocher
1c13e67b33 evolution bug fix #346 2020-07-09 22:08:42 -07:00
Glenn Jocher
17b9d7cdda
Merge pull request #344 from Laughing-q/master
update os.sep
2020-07-09 21:09:33 -07:00
Glenn Jocher
59beae98e8 add if isfile() to get_hash() 2020-07-09 20:57:24 -07:00
Glenn Jocher
6649770001 --rect test bug fix 2020-07-09 20:39:11 -07:00
Laughing
51a161d9e1
Update train.py 2020-07-10 11:37:58 +08:00
Glenn Jocher
d40e4781f1 --rect test bug patch 2020-07-09 20:36:30 -07:00
Laughing
41ab1b23f1
Merge pull request #1 from ultralytics/master
Master
2020-07-09 22:35:26 -05:00
Glenn Jocher
520f5de6f0 Label caching foundational re-write #306 2020-07-09 20:07:16 -07:00
Glenn Jocher
f310ca3bff Update issue templates 2020-07-09 17:36:26 -07:00
Glenn Jocher
e16e9e43e1 new nc=len(names) check 2020-07-09 17:10:43 -07:00
Glenn Jocher
cb527d3af9 new nc=len(names) check 2020-07-09 17:03:12 -07:00
Glenn Jocher
dd8e742ece
Merge pull request #334 from lorenzomammana/feature-multiple-datasets-training
Handle multiple datasets
2020-07-09 16:29:21 -07:00
Glenn Jocher
dd33d2ab77
Update datasets.py 2020-07-09 16:28:20 -07:00
Glenn Jocher
2110f5ee89 update log_dir to runs/exp #107 2020-07-09 16:03:10 -07:00
Glenn Jocher
603ea0bfdc update log_dir to runs/exp #107 2020-07-09 15:58:07 -07:00
Glenn Jocher
72d5b58b9a disable LR plot to suppress warning message 2020-07-09 15:16:57 -07:00
Glenn Jocher
24c5a941f0 --resume EMA fix #292 2020-07-09 15:09:06 -07:00
Glenn Jocher
2b6209a9d5
Merge pull request #338 from alexstoken/hyp_save_bugfix
Move hyp and opt yaml save to top of train()
2020-07-09 14:35:32 -07:00
Alex Stoken
9d631408a2
Move hyp and opt yaml save to top of train()
Fixes bug where scaled values were saved in hyp.yaml, which would cause continuity issues with --resume
2020-07-09 16:18:55 -05:00
Glenn Jocher
13a09fa68a
Update datasets.py 2020-07-09 13:47:20 -07:00
Glenn Jocher
47bf1730fa
Update datasets.py 2020-07-09 13:45:55 -07:00