comment about user variables

goktug97 · Apr 13, 2021 · 247cf19 · 247cf19
1 parent 64989cd
commit 247cf19
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/README.md b/README.md
@@ -85,6 +85,7 @@ user_config = Config(default_config)
 @user_config('optimizer')
 class UserOptimizerConfig():
     optim_type = torch.optim.Adam
+    # Users can add their own variables too
     amsgrad: Condition[bool] = Condition(False, default_config.optimizer.optim_type == torch.optim.Adam)
     nesterov: Condition[bool] = Condition(False, default_config.optimizer.optim_type == torch.optim.SGD)