2 Commits (a26e7de2bffaf5a87e7ed83aeabd0f0b2e8ad861)

Author SHA1 Message Date
  fcakyon 0e2d0d54d7
Fix typo (#3729) 3 years ago
  Glenn Jocher 1f69d12591
Update 4 main ops for paths and .run() (#3715) 3 years ago
  Glenn Jocher fad27c0046
Update DDP for `torch.distributed.run` with `gloo` backend (#3680) 3 years ago
  lb-desupervised bfb2276b1d
Slightly modify CLI execution (#3687) 3 years ago
  Glenn Jocher 53ed872c28
Update export.py, yolo.py `sys.path.append()` (#3579) 3 years ago
  Glenn Jocher 0e5cfdbea7
Refactor models/export.py arguments (#3564) 3 years ago
  Glenn Jocher f3c3d2ce5d
Merge `develop` branch into `master` (#3518) 3 years ago
  Yonghye Kwon 8d1ddc93c7
Earlier `assert` for cpu and half option (#3508) 3 years ago
  Yonghye Kwon a1c3572bc9
Explicit `model.eval()` call `if opt.train=False` (#3475) 3 years ago
  Sam_S 044daafd9d
Add `output_names` argument for ONNX export with dynamic axes (#3456) 3 years ago
  WangChaofeng ef4d53818d
ONNX export in .train() mode fix (#3362) 3 years ago
  Glenn Jocher 7b36e38cf8
Check CoreML models.train() mode (#3262) 3 years ago
  Cristi Fati 9ab561dbfc
Parameterize ONNX `--opset-version` (#3154) 3 years ago
  Cristi Fati d9b4e6b748
Add `--include torchscript onnx coreml` argument (#3137) 3 years ago
  Glenn Jocher 525f4f86a9
Add --optimize argument (#3093) 3 years ago
  Glenn Jocher e97d129db4
Update export.py with --train mode argument (#3066) 3 years ago
  Glenn Jocher 3ef3a95cfa
Do not optimize CoreML TorchScript model (#3055) 3 years ago
  jylink b292837e36
Fix ONNX export using --grid --simplify --dynamic simultaneously (#2982) 3 years ago
  Hodovo e2a80c6c0f
Add support for FP16 (half) to export.py (#3010) 3 years ago
  Glenn Jocher 4200674a13
Add yolov5/ to sys.path() for *.py subdir exec (#2949) 3 years ago
  Glenn Jocher 1b1ab4cca2
Add file_size() function (#2911) 3 years ago
  Glenn Jocher 646386ff09
Update export.py for 2 dry runs (#2910) 3 years ago
  Glenn Jocher a2a514dec8
Update export.py (#2909) 3 years ago
  Glenn Jocher 28db237639
Default optimize_for_mobile() on TorchScript models (#2908) 3 years ago
  Tim Stokman 1df8c6c963
Fix ONNX dynamic axes export support with onnx simplifier, make onnx simplifier optional (#2856) 3 years ago
  Glenn Jocher 1f3e482bce
ONNX Simplifier (#2815) 3 years ago
  Glenn Jocher 0f395b3e3b
YOLOv5 v5.0 Release patch 1 (#2764) 3 years ago
  Jan Hajek 7a0a81fd1d
GPU export options (#2297) 3 years ago
  Aditya Lohia 95aefea493
Dynamic ONNX engine generation (#2208) 3 years ago
  Glenn Jocher 6bd5e8bca7
nn.SiLU() export support (#1713) 3 years ago
  Glenn Jocher d45e349a1e
Rearrange export input after checks (#1118) 4 years ago
  Glenn Jocher 883924d9dc classifier, export, torch seed updates 4 years ago
  Glenn Jocher a12698f4c9
Update ci-testing.yml reflecting sys.path additions (#869) 4 years ago
  PetrDvoracek d55c326dda
export.py folder sys.path addition (#1052) 4 years ago
  Glenn Jocher d68179953d Update export.py 4 years ago
  Glenn Jocher 1e7674d4dd Update export.py --img-size comment 4 years ago
  Glenn Jocher 4fb8cb353f Update export.py with v3.0 Hardswish() support (#831) 4 years ago
  Glenn Jocher 4d7f222f73 Update export.py with v3.0 Hardswish() support 4 years ago
  Glenn Jocher a8751e50de model.fuse() fix for export.py (#827) 4 years ago
  Glenn Jocher b42e8a531b optimize imports 4 years ago
  NanoCode012 0892c44bc4
Fix Logging (#719) 4 years ago
  Jirka Borovec d5b6416c87
Explicit Imports (#498) 4 years ago
  Glenn Jocher 9006b85d21 update export.py torchscript file extension 4 years ago
  Laurentiu Diaconu a9918fbf2b updated coreml conversion to do pixel scaling between 0.0 and 1.0 4 years ago
  Glenn Jocher 659ad748c8 update get_voc.sh 4 years ago
  Glenn Jocher 7bd7b2c090 update export.py 4 years ago
  Glenn Jocher a9d20eba3e export.py update 4 years ago
  Glenn Jocher 569fbc63c8 export.py update 4 years ago
  Glenn Jocher a62a1c2c67 export.py update 4 years ago
  Glenn Jocher f1d67f4110 update export.py 4 years ago