Views
No views yet
src/scripts/train/main/csgo.sh| Key | Value |
|---|---|
| Architecture | NanoWM-L/2 |
| Dataset | csgo |
| Prediction | v |
| Noise schedule | squaredcos_cap_v2 (ZTSNR=True) |
| Steps | 50000 |
| Batch | 6/GPU |
1import sys
2from omegaconf import OmegaConf
3from safetensors.torch import load_file
4sys.path.insert(0, "src")
5from models import get_models
6
7cfg = OmegaConf.load("ckpt/config.yaml")
8cfg.experiment.infra.compile = False
9model = get_models(cfg).eval()
10state_dict = load_file("ckpt/model.safetensors")
11model.load_state_dict(state_dict, strict=True)