fixes requirements,code and readme

Author: sumantro93

run_python_examples.sh

@@ -195,9 +195,9 @@ function stop() {
     time_sequence_prediction/traindata.pt \
     word_language_model/model.pt \
     gcn/cora/ \
-    gat/cora/ || error "couldn't clean up some files"
-    swin_transformer/swin_cifar10.pt || error "couldn't clean up some files"
-
+    gat/cora/ || error "couldn't clean up some files" \
+    swin_transformer/ \
+    swin_trasformer/swin_cifar10.pt || error "command swin_transformer/swin_cifar10.pt not found" \
   git checkout fast_neural_style/images/output-images/amber-candy.jpg || error "couldn't clean up fast neural style image"
 
   base_stop "$1"
@@ -225,7 +225,7 @@ function run_all() {
   run fx
   run gcn
   run gat
-  run swin
+  run swin_transformer
 }
 
 # by default, run all examples

swin_transformer/README.md

@@ -32,10 +32,10 @@ pip install -r requirements.txt
 
 ## Usage
 
-### Train the model
+### Train & Save the model
 
 ```bash
-python swin_transformer.py --epochs 10 --batch-size 64 --lr 0.001
+python swin_transformer.py --epochs 10 --batch-size 64 --lr 0.001 --save-model
 ```
 
 ### Test the model
@@ -44,13 +44,7 @@ Testing is done automatically after each epoch. To only test, run with:
 
 ```bash
 python swin_transformer.py --epochs 1
-```
-
-### Save the model
-
-```bash
-python swin_transformer.py --save-model
-```
+``
 
 The model will be saved as `swin_cifar10.pt`.
 

swin_transformer/requirements.txt

@@ -1,2 +1,2 @@
-torch
+torch>=2.6
 torchvision

swin_transformer/swin_transformer.py

@@ -1,6 +1,4 @@
-from __future__ import print_function
 import argparse
-
 import torch
 import torch.nn as nn
 import torch.nn.functional as F
@@ -202,6 +200,4 @@ def main():
 
     if args.save_model:
         torch.save(model.state_dict(), "swin_cifar10.pt")
-
-if __name__ == '__main__':
-    main()
+main()