File size: 267 Bytes
9b896f5
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
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")