Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
VAR_pytorch
Commits
87180d7d
Commit
87180d7d
authored
Nov 29, 2024
by
suily
Browse files
Update README.md
parent
9f70f93b
Pipeline
#2011
failed with stages
in 0 seconds
Changes
1
Pipelines
2
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
0 additions
and
12 deletions
+0
-12
README.md
README.md
+0
-12
No files found.
README.md
View file @
87180d7d
...
@@ -70,18 +70,6 @@ pip install -r requirements.txt
...
@@ -70,18 +70,6 @@ pip install -r requirements.txt
git config --global --add safe.directory /your_code_path/VAR
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`
`imagenet-1k`
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment