- generate training instances by running gen_ins.py. You can change the pagerank settings in the bottom of the code.
- run train.py to train the model, you can check both trianing and valid loss on primal/dual MSE loss
CenturionMKIII/PDHG-Net-New
Folders and files
| Name | Name | Last commit date | ||
|---|---|---|---|---|