Fixed allow_pickle and multi-gpu issues - #8

Open

wuyenlin wants to merge 2 commits into

sunnychencool:masterfrom

wuyenlin:master

Commits on Jun 30, 2021

allow pickle
wuyenlin
committed
6857b24
allow pickle and added device list for single and multi gpu setup
wuyenlin
committed
fdac9e8