Commit ·
abac145
1
Parent(s): f3a2b23
Update modeling_vivqa.py
Browse files- modeling_vivqa.py +1 -1
modeling_vivqa.py
CHANGED
|
@@ -37,7 +37,7 @@ class Blip2EfficientExtractor(nn.Module):
|
|
| 37 |
self.model_blip2.eval()
|
| 38 |
|
| 39 |
# Efficientnet
|
| 40 |
-
self.model_efficient = EfficientNet.from_pretrained('efficientnet-b7').to(self.device)
|
| 41 |
self.model_efficient.eval()
|
| 42 |
self.pooling1 = nn.AdaptiveAvgPool2d((1, 32))
|
| 43 |
self.pooling2 = nn.AdaptiveAvgPool2d((1, 768))
|
|
|
|
| 37 |
self.model_blip2.eval()
|
| 38 |
|
| 39 |
# Efficientnet
|
| 40 |
+
self.model_efficient = EfficientNet.from_pretrained('efficientnet-b7', advprop=True).to(self.device)
|
| 41 |
self.model_efficient.eval()
|
| 42 |
self.pooling1 = nn.AdaptiveAvgPool2d((1, 32))
|
| 43 |
self.pooling2 = nn.AdaptiveAvgPool2d((1, 768))
|