diff --git a/evo_kit/README.md b/evo_kit/README.md index ecac85379c048f22e444b9286d8e5225a7e7daa8..4305c572ab233b5f17ed3f849a551276059f0c69 100644 --- a/evo_kit/README.md +++ b/evo_kit/README.md @@ -22,9 +22,7 @@ agent->update(noisy_info, noisy_rewards); ``` ## 一键运行demo列表 -- **PaddleLite**: sh ./scripts/build.sh paddle -- **Torch**: sh ./scripts/build.sh torch -- **裸写网络**: +- sh ./scripts/build.sh ## 相关依赖: - Protobuf2 diff --git a/evo_kit/scripts/build_torch_demo.sh b/evo_kit/scripts/build.sh similarity index 96% rename from evo_kit/scripts/build_torch_demo.sh rename to evo_kit/scripts/build.sh index b2f4df4444012c49bade049a7b30c9ebf637cafb..fbc68c08d9b883e15fc793a54f182f55351f23b3 100644 --- a/evo_kit/scripts/build_torch_demo.sh +++ b/evo_kit/scripts/build.sh @@ -1,5 +1,7 @@ #!/bin/bash +sh scripts/lib_install.sh torch + cd demo/torch #---------------libtorch-------------#