chore: 简化格式并更新文档

khaosz/trainer/strategy.py

@@ -7,7 +7,7 @@ import torch.nn.functional as F
 from torch.nn.parallel import DistributedDataParallel as DDP
 
 from torch import Tensor
-from typing import Any, Callable, Dict, Union, Optional
+from typing import Any, Callable, Dict, Union
 from abc import ABC, abstractmethod
 
 

khaosz/trainer/train_callback.py

@@ -6,7 +6,6 @@ import torch.nn as nn
 from pathlib import Path
 from tqdm import tqdm
 from torch.nn.utils import clip_grad_norm_
-from torch.optim.lr_scheduler import LRScheduler
 from typing import Callable, List, Optional, Protocol
 
 from khaosz.parallel import only_on_rank