Spaces:
Sleeping
Sleeping
import torch | |
from pathlib import Path | |
dir=Path.home() / f"tmp/resnet50/CUB2011/123456/" | |
dic=torch.load(dir/ f"SlDD_Selection_50.pt") | |
print (dic) | |
#if 'linear.selection' in dic.keys(): | |
#print("key 'linear.selection' exist") | |
#else: | |
#print("no such key") | |