filter the args

Author: exx8

train.py

@@ -734,9 +734,8 @@ def main():
     if should_log_to_tensorboard(args):
         if has_tensorboard:
             tensorboard_writer = SummaryWriter(args.log_tensorboard)
-            #write Hyperparameters to tensorboard
-            #get all args keys that are  one of int, float, str, bool, or torch.Tensor
-            hparams = {k: v for k, v in vars(args).items() if type(v) in [int, float, str, bool, torch.Tensor]}
+            #write Hyperparameters to tensorboard, remove Nones
+            hparams = {k: v for k, v in vars(args).items() if v is not None}
             tensorboard_writer.add_hparams(hparams, {})
 
         else: