21 lines
888 B
Python
21 lines
888 B
Python
|
import torch
|
|||
|
|
import pickle
|
|||
|
|
import glob
|
|||
|
|
|
|||
|
|
label = "gait-conditioned-agility/pretrain-go2/train"
|
|||
|
|
dirs = glob.glob(f"../../runs/{label}/*")
|
|||
|
|
logdir = sorted(dirs)[0]
|
|||
|
|
|
|||
|
|
# transfer
|
|||
|
|
with open(logdir+"/parameters.pkl", 'rb') as file:
|
|||
|
|
pkl_cfg = pickle.load(file)
|
|||
|
|
# 我们通过 torch.is_tensor(v) 检查字典中的每个值 v 是否为 PyTorch 张量。
|
|||
|
|
# 如果是，我们应用 .cpu() 方法将其转移到 CPU；如果不是，我们保留原值。
|
|||
|
|
# 这样，只有真正的张量会被转移，其他类型的值（如整数、字符串等）会保持不变。
|
|||
|
|
pkl_cfg_cpu = {k: v.cpu() if torch.is_tensor(v) else v for k, v in pkl_cfg.items()}
|
|||
|
|
print("Transfer Succeed ! !")
|
|||
|
|
|
|||
|
|
# save transferred .pkl file
|
|||
|
|
with open(logdir+"/parameters_cpu.pkl", 'wb') as file:
|
|||
|
|
pickle.dump(pkl_cfg_cpu, file)
|
|||
|
|
print("Transferred Pickle File has been saved as parameters_cpu.pkl")
|