Update 'tac_ssl_tag.py'

2023-09-12 14:07:39 +00:00 · 2023-09-12 14:07:39 +00:00 · 5634afcfcd
commit 5634afcfcd
parent 913e000336
1 changed files with 1 additions and 0 deletions
--- a/tac_ssl_tag.py
+++ b/tac_ssl_tag.py
@ -318,6 +318,7 @@ if __name__ == "__main__":
        # 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()) + \
                 list(model.Phi_vision_q.parameters()) + list(model.Phi_tactile_q.parameters())
+        optimizer = optim.Adam(modules, lr=0.03)
        classifier, criterion = None, None
    else:
        # Initialize training