onnxruntime
Refactor device handling and basic support for PyTorch Lightning
#6758

Merged

Refactor device handling and basic support for PyTorch Lightning #6758

thiagocrepaldi merged 4 commits into thiagofc/ortmodule-api from thiagofc/support-lightning

thiagocrepaldi requested a review from

baijumeswani 5 years ago

thiagocrepaldi requested a review from

BowenBao 5 years ago

thiagocrepaldi requested a review from

liqunfu 5 years ago

thiagocrepaldi requested a review from

spandantiwari 5 years ago

thiagocrepaldi requested a review 5 years ago

thiagocrepaldi force pushed 5 years ago

thiagocrepaldi force pushed to a85a8891 5 years ago

Move device change detection to forward method

8e7f1ea9

Add PyTorch lightning sample to CI

83a49f93

thiagocrepaldi force pushed from a85a8891 to 83a49f93 5 years ago

baijumeswani commented on 2021-02-23

Add UT that wraps ORTModule and change devices

ab0e9b92

baijumeswani commented on 2021-02-24

Address comments

fca1cc2a

thiagocrepaldi force pushed from 41bedeb7 to fca1cc2a 5 years ago

baijumeswani approved these changes on 2021-02-24

thiagocrepaldi merged aa5cd37a into thiagofc/ortmodule-api 5 years ago

thiagocrepaldi deleted the thiagofc/support-lightning branch 5 years ago

thiagocrepaldi added component:ortmodule

thiagocrepaldi added component:training-frontend

Reviewers

baijumeswani

BowenBao

liqunfu

spandantiwari

Assignees

No one assigned

Labels

None yet

Milestone

No milestone