add doc

Signed-off-by: Can-Zhao <[email protected]>

Author: Can-Zhao

docs/source/networks.rst

@@ -753,6 +753,7 @@ Utilities
 
 Noise Schedulers
 ----------------
+.. automodule:: monai.networks.schedulers
 .. currentmodule:: monai.networks.schedulers
 
 `Scheduler`

monai/networks/schedulers/rectified_flow.py

@@ -71,7 +71,7 @@ class RFlowScheduler(Scheduler):
     Supports uniform and logit-normal sampling methods, timestep transformation for
     different resolutions, and noise addition during diffusion.
 
-    Attributes:
+    Args:
         num_train_timesteps (int): Total number of training timesteps.
         use_discrete_timesteps (bool): Whether to use discrete timesteps.
         sample_method (str): Training time step sampling method ('uniform' or 'logit-normal').