Skip to content
Snippets Groups Projects

Passed device to oxford_vgg_resnets

Open Manuel Günther requested to merge vgg2-gpu into master
1 file
+ 8
2
Compare changes
  • Side-by-side
  • Inline
@@ -1193,7 +1193,11 @@ def afffe_baseline(
@@ -1193,7 +1193,11 @@ def afffe_baseline(
def oxford_vgg2_resnets(
def oxford_vgg2_resnets(
model_name, annotation_type, fixed_positions=None, memory_demanding=False
model_name,
 
annotation_type,
 
fixed_positions=None,
 
memory_demanding=False,
 
device=torch.device("cpu"),
):
):
"""
"""
Get the pipeline for the resnet based models from Oxford.
Get the pipeline for the resnet based models from Oxford.
@@ -1227,7 +1231,9 @@ def oxford_vgg2_resnets(
@@ -1227,7 +1231,9 @@ def oxford_vgg2_resnets(
transformer = embedding_transformer(
transformer = embedding_transformer(
cropped_image_size=cropped_image_size,
cropped_image_size=cropped_image_size,
embedding=OxfordVGG2Resnets(
embedding=OxfordVGG2Resnets(
model_name=model_name, memory_demanding=memory_demanding
model_name=model_name,
 
memory_demanding=memory_demanding
 
device=device
),
),
cropped_positions=cropped_positions,
cropped_positions=cropped_positions,
fixed_positions=fixed_positions,
fixed_positions=fixed_positions,
Loading