Skip to content

Commit 5586c3b

Browse files
committed
add swin transformer example
1 parent 7d93abd commit 5586c3b

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

run_python_examples.sh

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -164,6 +164,10 @@ function gat() {
164164
uv run main.py --epochs 1 --dry-run || error "graph attention network failed"
165165
}
166166

167+
function swin() {
168+
uv run swin_transformer.py --epochs 1 --dry-run || error "swin transformer failed"
169+
}
170+
167171
eval "base_$(declare -f stop)"
168172

169173
function stop() {
@@ -189,6 +193,7 @@ function stop() {
189193
word_language_model/model.pt \
190194
gcn/cora/ \
191195
gat/cora/ || error "couldn't clean up some files"
196+
swin_transformer/swin_cifar10.pt || error "couldn't clean up some files"
192197

193198
git checkout fast_neural_style/images/output-images/amber-candy.jpg || error "couldn't clean up fast neural style image"
194199

@@ -217,6 +222,7 @@ function run_all() {
217222
run fx
218223
run gcn
219224
run gat
225+
run swin
220226
}
221227

222228
# by default, run all examples

0 commit comments

Comments
 (0)