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

Merged

Commits

Preload the data to train_data.

xuzhao9 committed 5 years ago
Copy data to device in `__init__()`.

xuzhao9 committed 5 years ago
Remove redundant cuda() calls in train().

xuzhao9 committed 5 years ago
Fix CI error.

xuzhao9 committed 5 years ago
Fix another CI issue.

xuzhao9 committed 5 years ago
Fix cuda issue.

xuzhao9 committed 5 years ago
Copy data to cuda in train().

xuzhao9 committed 5 years ago

Loading