3 lines
62 B
Diff
3 lines
62 B
Diff
|
27d26
|
||
|
|
< assert eles[0] in ['cpu', 'cuda', 'gpu'], err_msg
|