UGM Models

[UGM (Utilities for Ground Motions)]的预训练模型文件仓库。

文件

文件	大小	用途
KNET-StyleGAN-fid1.730.pkl	~309 MB	StyleGAN3 条件生成器（KNET 数据集训练，FID 1.73），用于地震动时频谱生成
vgg16.pt	~528 MB	TorchScript 格式的 VGG16 LPIPS 感知损失模型，用于潜空间反演

下载

# 方式一：UGM 内置脚本（推荐）
pip install -e ".[hf]"
python -m ugm.download_models

# 方式二：huggingface_hub
python -c "from huggingface_hub import snapshot_download; snapshot_download('xzk8559/UGM', local_dir='~/.cache/ugm/models')"

许可

模型权重遵循 CC BY-NC 4.0（署名-非商业性使用）。代码仓库见 UGM 主仓库。 vgg16.pt 原始来源为 NVIDIA StyleGAN2 的 LPIPS 权重（NVIDIA 许可）。