LibCity / Bigscity-LibCity

LibCity: An Open Library for Urban Spatial-temporal Data Mining
https://libcity.ai/
Apache License 2.0
901 stars 165 forks source link

关于保存好的模型如何使用 #400

Closed hybridIce closed 6 months ago

hybridIce commented 7 months ago

老师你好,我现在已经训练完成了模型,准备画attention热力图了,我看模型被保存成一个压缩文件了,请问libcity提供这个运行已保存好的模型的功能吗

Kazeya27 commented 7 months ago

将--train参数设置为false,--exp_id参数设置已训练模型的exp_id即可