How can resume the train_search.py using weights in log_dir if the code stops due to accidents like power cut ... ?

[Li Jiahui]

[chenxin061]

You can resue it with regular PyTorch functions such as torch.load() and model.load_state_dict().