Commit 87180d7d authored by suily's avatar suily
Browse files

Update README.md

parent 9f70f93b
Pipeline #2011 failed with stages
in 0 seconds
......@@ -70,18 +70,6 @@ pip install -r requirements.txt
git config --global --add safe.directory /your_code_path/VAR
```
### 代码改动说明
ps:仓库中是改动后的代码,不需再次修改
```
# 1、VAR/train.py
is_val_and_also_saving = (ep + 1) % 1 == 0 or (ep + 1) == args.ep # 训练时间过长,为防止中断修改为每轮保存一次模型,原为(ep + 1) % 10
vae_ckpt = '/path/your_code_data/VAR/vae_ch160v4096z32.pth' # 修改vae模型位置
# 2、VAR/models/basic_var.py
fused_mlp_func = None # import下添加,dcu不支持 fused_dense.linear_act_forward
memory_efficient_attention = None # import下添加,对于该仓库xformers没有适合dcu的算子
```
## 数据集
`imagenet-1k`
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment