3. light wam 模型加载
模型加载与初始化
import os
import hydra
import torch
from omegaconf import DictConfig, OmegaConf
from hydra.utils import instantiate
from torch.utils.data import DataLoaderfrom lightwam.runtime import _resolve_train_device, _normalize_mixed_precision, _mix…