TUAN-ANH

Looks like pytorch mobile doesn't support cuda yet. so when converting weights, don't use .cuda() commands for inputs or models.