Skip to content
This repository was archived by the owner on Jun 3, 2025. It is now read-only.

Conversation

@bfineran
Copy link
Contributor

  • option to disable fp16 in second test loop
  • unwrap DDP model before setting export mode
  • image size param update

@bfineran bfineran requested a review from a team March 24, 2021 22:58
@bfineran bfineran self-assigned this Mar 24, 2021
@bfineran bfineran requested a review from a team March 24, 2021 23:39
@bfineran bfineran merged commit 6220d31 into main Mar 25, 2021
@bfineran bfineran deleted the bfineran-yolo-train-patch-1 branch March 25, 2021 00:51
bfineran added a commit that referenced this pull request Mar 25, 2021
* Update train.py

* update onnx export for DDP
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants