Update README.md

LukasHedegaard · Jun 21, 2021 · edda3df · edda3df
1 parent 53fbb5a
commit edda3df
Showing 1 changed file with 21 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -1 +1,21 @@
-# OadTR
+# OadTR
+Code for our paper: "OadTR: Online Action Detection with Transformers"
+
+## Dependencies
+
+* Pytorch==1.6.0 
+* json
+* numpy
+* tensorboard-logger
+* torchvision==0.7.0
+
+# Feature is coming soon...
+
+# Training
+```
+python main.py --num_layers 3 --decoder_layers 5 --enc_layers 64 --output_dir models/en_3_decoder_5_lr_drop_1
+```
+# Validation
+```
+python main.py --num_layers 3 --decoder_layers 5 --enc_layers 64 --output_dir models/en_3_decoder_5_lr_drop_1 --eval --resume models/en_3_decoder_5_lr_drop_1/checkpoint000{}.pth
+```