Yugeng Liu
Results
2
comments of
Yugeng Liu
No, you can't directly return any weights because the DD code splits the weights into a list. I think this is because they use a very early version of PyTorch.
I think the parameters are saved in the ```results.pth```. You can load it as the ```step```. For instance, ``` steps = torch.load(os.path.join(data_path, 'results.pth')) steps = [(d.to(state.device), l.to(state.device), lr.to(state.device)) for (d,...