brian/visual_critical_discourse_analysis
1
0
Fork 0
Code Issues 11 Pull Requests Actions Projects 1 Releases Activity

added option to download weights

Browse Source
This commit is contained in:
Brian Bjarke Jensen
2024-07-28 21:25:24 +02:00
parent 6541a47668
commit be6833ac6a
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
model/src/models/visual_communication.py
+2 -2
View File
@@ -35,10 +35,10 @@ from .visual_syntax import VisualSyntaxTail
class VisualCommunicationModel(nn.Module): class VisualCommunicationModel(nn.Module):
"""Visual communication model.""" """Visual communication model."""
def __init__(self): def __init__(self, download_resnet_weights: bool = False):
super().__init__() super().__init__()
# store other models # store other models
self.resnet_head = ResNet18Head() self.resnet_head = ResNet18Head(download_resnet_weights)
self.visual_syntax_tail = VisualSyntaxTail() self.visual_syntax_tail = VisualSyntaxTail()
self.contact_tail = ContactTail() self.contact_tail = ContactTail()
self.angle_tail = AngleTail() self.angle_tail = AngleTail()