Skip to content
Snippets Groups Projects
Commit a558259a authored by Guillaume HEUSCH's avatar Guillaume HEUSCH
Browse files

[extractor] set pretrained FASnet as False

parent ee78d11d
No related branches found
No related tags found
1 merge request!28Resolve "Tests take forever to run"
Pipeline #28512 passed
......@@ -44,7 +44,7 @@ class FASNetExtractor(Extractor):
# model
self.transforms = transforms
self.network = FASNet()
self.network = FASNet(pretrained=False)
#self.network=self.network.to(device)
......@@ -93,4 +93,4 @@ class FASNetExtractor(Extractor):
# output is a scalar score
return output
\ No newline at end of file
return output
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment