用了下LRUCache
from LRUCache import LRUCache
import subprocess
def get_model(model_path):
print("get model")
if model_path == "":
return None;
model = cache.get(model_path)
if model!=-1:
return model
local_path = head+model_path.split("/")[-1]
print(f"input model_path:{model_path}\nlocal_path:{local_path}")
print(1)
cmd_ret = None
print(os.path.exists(local_path))
if not os.path.exists(local_path):
index_cmd_return = subprocess.run(["./azcopy", "copy", url_template.format(model_path),head,"--recursive"])
model = load_model(local_path)
print(2)
cache.set(model_path,model)
return model
get_model("ao/als/aia-tra/b/model_epoch5-it235-loss1.514")