[BUG] Error on hypopt pipeline test when device using cuda

nodefluxio / vortex

A Deep Learning Model Development Framework for Computer Vision

27 stars 6 forks source link

[BUG] Error on hypopt pipeline test when device using cuda #31

Closed alphinside closed 4 years ago

alphinside commented 4 years ago

Describe the bug Error on hypopt pipeline test when device using cuda caused by the returned array from training pipeline is an array of cuda tensor

To Reproduce Steps to reproduce the behavior:

update tests/config/test_classification_pipelines.yml device to 'cuda'

Expected behavior No error occured