Minor changes

alex-snd · Sep 6, 2022 · 9531129 · 9531129
1 parent a625460
commit 9531129
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 5 deletions.
diff --git a/src/trecover/train/collab/entrypoint.py b/src/trecover/train/collab/entrypoint.py
@@ -21,8 +21,6 @@
 def monitor(cli_args: Optional[List[str]] = None) -> None:
     args = arguments.sync_base_args(arguments.get_monitor_parser().parse_args(cli_args))
 
-    args.sync_args = True  # TODO
-
     if args.assist_in_averaging and args.client_mode:
         log.project_console.print('Client-mode peers cannot assist in averaging', style='red')
         return
@@ -82,8 +80,6 @@ def monitor(cli_args: Optional[List[str]] = None) -> None:
 def train(cli_args: Optional[List[str]] = None) -> None:
     args = arguments.sync_base_args(arguments.get_train_parser().parse_args(cli_args))
 
-    args.sync_args = True  # TODO
-
     os.system('ulimit -n 16384')
 
     if args.batch_size is None:

diff --git a/src/trecover/train/collab/monitor.py b/src/trecover/train/collab/monitor.py
@@ -220,6 +220,6 @@ def _print_metrics(step: int, metrics: GlobalMetrics) -> None:
     def _upload_state(self) -> None:
         if self._is_time_to_upload:
             wandb.save(str(self.aux_opt.state_path.absolute()),
-                       base_path=str(self.aux_opt.state_path.parent),  # TODO check, change name, delete at max files
+                       base_path=str(self.aux_opt.state_path.parent),
                        policy='now')
             self.last_upload_time = time.monotonic()