Create a custom predict function

projects
2

Yes, you can access the .weight attribute and transform them to numpy arrays e.g. via:

arr = model.layer.weight.cpu().numpy()