From 5634afcfcd3b56232e648397f52ec424fc9a5a82 Mon Sep 17 00:00:00 2001 From: Vedant Dave Date: Tue, 12 Sep 2023 14:07:39 +0000 Subject: [PATCH] Update 'tac_ssl_tag.py' --- tac_ssl_tag.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tac_ssl_tag.py b/tac_ssl_tag.py index 69159a3..712d51e 100755 --- 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