search

boostcampaitech2 / image-classification-level1-08

image-classification-level1-08 3rd Place!
7 stars 3 forks source link

pickle save #33

Open jinmang2 opened 3 years ago

jinmang2 commented 3 years ago 
import pickle

with open("logits.pkl", "wb") as f:
    pickle.dump(logits, f, protocol=pickle.HIGHEST_PROTOCOL)

with open("logits.pkl", "rb") as f:
    z = pickle.load(f)