File tree Expand file tree Collapse file tree 1 file changed +6
-0
lines changed Expand file tree Collapse file tree 1 file changed +6
-0
lines changed Original file line number Diff line number Diff line change @@ -164,6 +164,10 @@ function gat() {
164
164
uv run main.py --epochs 1 --dry-run || error " graph attention network failed"
165
165
}
166
166
167
+ function swin() {
168
+ uv run swin_transformer.py --epochs 1 --dry-run || error " swin transformer failed"
169
+ }
170
+
167
171
eval " base_$( declare -f stop) "
168
172
169
173
function stop() {
@@ -189,6 +193,7 @@ function stop() {
189
193
word_language_model/model.pt \
190
194
gcn/cora/ \
191
195
gat/cora/ || error " couldn't clean up some files"
196
+ swin_transformer/swin_cifar10.pt || error " couldn't clean up some files"
192
197
193
198
git checkout fast_neural_style/images/output-images/amber-candy.jpg || error " couldn't clean up fast neural style image"
194
199
@@ -217,6 +222,7 @@ function run_all() {
217
222
run fx
218
223
run gcn
219
224
run gat
225
+ run swin
220
226
}
221
227
222
228
# by default, run all examples
You can’t perform that action at this time.
0 commit comments