Update README.md

HKUDS · Feb 22, 2023 · ad5ea1c · ad5ea1c
1 parent 05a4dc0
commit ad5ea1c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/README.md b/README.md
@@ -4,7 +4,7 @@
 This is the implementation of Automated Spatio-Temporal Graph Contrastive Learning (AutoST):
 
 ## Requirements ##
-Pytorch = 1.7.0 and Tensorflow = 1.15.3 (crime prediction task (ST-SHN))
+python = 3.7.13, Pytorch = 1.12.0 with cuda = 10.2  and Tensorflow = 1.15.3 (crime prediction task (ST-SHN))
 
 ## Data ##
 Moblity data, crime data, census block data, POI data, house price of Chicago ([https://data.cityofchicago.org/](https://data.cityofchicago.org/)) and New York([https://opendata.cityofnewyork.us/](https://opendata.cityofnewyork.us/)).