feat: added logging info to tensorboard writer

token-tricksters · Jun 26, 2023 · ecd4c6e · ecd4c6e
1 parent 7079998
commit ecd4c6e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/classifier.py b/classifier.py
@@ -232,8 +232,6 @@ def save_model(model, optimizer, args, config, filepath):
 
 
 def train(args):
-    name = datetime.now().strftime("%Y%m%d-%H%M%S")
-    writer = SummaryWriter(log_dir=args.logdir + "/classifier/" + name)
     loss_idx_value = 0
 
     device = torch.device('cuda') if args.use_gpu else torch.device('cpu')
@@ -266,6 +264,9 @@ def train(args):
     optimizer = AdamW(model.parameters(), lr=lr)
     best_dev_acc = 0
 
+    name = f"{datetime.now().strftime('%Y%m%d-%H%M%S')}-lr={lr}-optimizer={type(optimizer).__name__}"
+    writer = SummaryWriter(log_dir=args.logdir + "/classifier/" + name)
+
     # Run for the specified number of epochs
     for epoch in range(args.epochs):
         model.train()

diff --git a/multitask_classifier.py b/multitask_classifier.py
@@ -132,8 +132,6 @@ def save_model(model, optimizer, args, config, filepath):
 
 ## Currently only trains on sst dataset
 def train_multitask(args):
-    name = datetime.now().strftime("%Y%m%d-%H%M%S")
-    writer = SummaryWriter(log_dir=args.logdir + "/multitask_classifier/" + name)
     loss_sst_idx_value = 0
     loss_sts_idx_value = 0
     loss_para_idx_value = 0
@@ -189,6 +187,9 @@ def train_multitask(args):
     best_dev_acc_sst = 0
     best_dev_acc_sts = 0
 
+    name = f"{datetime.now().strftime('%Y%m%d-%H%M%S')}-lr={lr}-optimizer={type(optimizer).__name__}"
+    writer = SummaryWriter(log_dir=args.logdir + "/multitask_classifier/" + name)
+
     # Run for the specified number of epochs
     for epoch in range(args.epochs):
         model.train()