VedantDave/TVSSL
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update 'tac_ssl_tag.py'

Browse Source
This commit is contained in:
Vedant Dave 2023-09-12 14:07:39 +00:00
parent 913e000336
commit 5634afcfcd
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tac_ssl_tag.py
View File

@ -318,6 +318,7 @@ if __name__ == "__main__":
# Initialize optimizer # Initialize optimizer
modules = list(model.vision_base_q.parameters()) + list(model.tactile_base_q.parameters()) + list(model.phi_vision_q.parameters()) + list(model.phi_tactile_q.parameters()) + \ modules = list(model.vision_base_q.parameters()) + list(model.tactile_base_q.parameters()) + list(model.phi_vision_q.parameters()) + list(model.phi_tactile_q.parameters()) + \
list(model.Phi_vision_q.parameters()) + list(model.Phi_tactile_q.parameters()) list(model.Phi_vision_q.parameters()) + list(model.Phi_tactile_q.parameters())
optimizer = optim.Adam(modules, lr=0.03)
classifier, criterion = None, None classifier, criterion = None, None
else: else:
# Initialize training # Initialize training