Passed device to oxford_vgg_resnets
Compare changes
@@ -1193,7 +1193,11 @@ def afffe_baseline(
@@ -1193,7 +1193,11 @@ def afffe_baseline(
@@ -1227,7 +1231,9 @@ def oxford_vgg2_resnets(
@@ -1227,7 +1231,9 @@ def oxford_vgg2_resnets(
While for all models, we are passing the device
, for the oxford models this is still missing. This MR solves this.