benchmark
Preload the data to train_data in init for Background_Matting
#396

Merged

Preload the data to train_data in init for Background_Matting #396

xuzhao9 merged 7 commits into pytorch:master from xuzhao9:xz9/fix-background-matting

Preload the data to train_data.

1d8552e9

facebook-github-bot added cla signed

xuzhao9 requested a review from

wconstab 4 years ago

xuzhao9 changed the title ~~Preload the data to train_data in __init__~~ Preload the data to train_data in __init__ for Background_Matting 4 years ago

wconstab approved these changes on 2021-06-09

Copy data to device in `__init__()`.

c444aabe

Remove redundant cuda() calls in train().

acf11163

Fix CI error.

c9c55816

Fix another CI issue.

96967cc0

xuzhao9 force pushed from 25eeae71 to 96967cc0 4 years ago

Fix cuda issue.

3b170f55

Copy data to cuda in train().

08f15bb8

xuzhao9 merged 5726bd71 into master 4 years ago

xuzhao9 deleted the xz9/fix-background-matting branch 4 years ago

Reviewers

wconstab

Assignees

No one assigned

Labels

cla signed

Milestone

No milestone

benchmark Preload the data to train_data in __init__ for Background_Matting #396 Merged

Preload the data to train_data in __init__ for Background_Matting #396

benchmark
Preload the data to train_data in init for Background_Matting
#396

Merged

Preload the data to train_data in init for Background_Matting #396