torchrun --nnodes=1 --nproc_per_node=N train.py --model DiT-XL/2 --data-path /path/to/imagenet/train We've trained DiT-XL/2 and DiT-B/4 models from scratch with the PyTorch training script to verify that it reproduces the original JAX results up to several hundred thousand traini...
Pytorchで再現性を制御するためにseedをセット. config=OmegaConf.load(f"{opt.config}")model=load_model_from_config(config,f"{opt.ckpt}") OmegaConfは設定系を扱いやすくするためのライブラリ. modelのロードはckptファイルからtorch.loadで読み込み device=torch.device("cuda")iftorch.cuda.is_...