jeffatllatos

recognition/__init__.py from 'cuda' to device parameter

1

Can you change this line only so we can use CPU or GPU. FROM: model = Backbone(num_layers=50, drop_ratio=0.6, mode='ir_se').to(**'cuda'**).eval() TO: model = Backbone(num_layers=50, drop_ratio=0.6, mode='ir_se').to(**device**).eval()