Fix bugs in LFNA

exps/LFNA/basic-maml.py

@@ -41,10 +41,7 @@ class MAML:
         )
         self.meta_lr_scheduler = torch.optim.lr_scheduler.MultiStepLR(
             self.meta_optimizer,
-            milestones=[
-                int(epochs * 0.8),
-                int(epochs * 0.9),
-            ],
+            milestones=[int(epochs * 0.8), int(epochs * 0.9),],
             gamma=0.1,
         )
         self.inner_lr = inner_lr
@@ -197,10 +194,7 @@ if __name__ == "__main__":
         help="The synthetic enviornment version.",
     )
     parser.add_argument(
-        "--hidden_dim",
-        type=int,
-        default=16,
-        help="The hidden dimension.",
+        "--hidden_dim", type=int, default=16, help="The hidden dimension.",
     )
     parser.add_argument(
         "--meta_lr",
@@ -230,16 +224,10 @@ if __name__ == "__main__":
         help="The gap between prev_time and current_timestamp",
     )
     parser.add_argument(
-        "--meta_batch",
-        type=int,
-        default=64,
-        help="The batch size for the meta-model",
+        "--meta_batch", type=int, default=64, help="The batch size for the meta-model",
     )
     parser.add_argument(
-        "--epochs",
-        type=int,
-        default=2000,
-        help="The total number of epochs.",
+        "--epochs", type=int, default=2000, help="The total number of epochs.",
     )
     parser.add_argument(
         "--early_stop_thresh",