-
2023/08/04
Add model overview.
-
2023/06/24
Add the train.py file.
This is a deep-learning-based model used to predict the location of congestion. We achieved an SSIM score of 0.863. During the prediction process, we incorporate the macro region feature, and Rectangular Uniform Wire Density (RUDY)[1], which serves as an early estimation of routing demand after placement. and also consider the RUDY pins, which are calculated based on individual pins and the nets connected to them.
- python3.8
- scipy
- matplotlib
- numpy
- pandas
- pytorch 1.12.0
- tqdm
- pytorch_msssim
python train.py--root_path: The path of the data file
--batch_size: The batch size figure
--num_epochs: The training epochs
--learning_rate: learning rate [0,1]
--weight_path: The path to save the model weight
--fig_path: The path of the figure file
python predict.py--data_path: The path of the data file
--fig_save_path: The path you want to save figure
--weight_path: The path of the model weight
--output_path: The path of the predict output with .npy file
--congestion_threshold: congestion_threshold [0,1]
--device: If you want to use gpu type "cuda"
- Tune your own congestion_threshold, the defalt is 0.5 as shown in following figure.
- The output coordinate csv file and image array npy file are in the ./output file.
- The model predict cost time is 0.71 ~ 1.8 sec.
- Congestion value:
$normalize(\frac{overflow}{totaltracks})$ .
SSIM score: 0.863
by pytorch_msssim.SSIM
[1] P. Spindler and F. M. Johannes, "Fast and accurate routing demand estimation for efficient routability-driven placement," *Design, Automation & Test in Europe Conference & Exhibition*, pp. 1-6, 2007.