ITADN

Questions about tcn checkpoint parameter count and conversion to PyTorch

#510Closedlulucai435 创建于 2025-08-09
L
lulucai435commented
Hi, I have some questions regarding the TCN model checkpoint: I converted the MLP part of TCN to PyTorch, but when saving the checkpoint from the original TCN version, I found the total parameter count is 18,432. According to the network architecture (147 → 64 → 64 → 64 → 4), the expected parameter count should be 18,052. So, there are about 380 extra parameters — could you please clarify what these extra parameters represent? Regarding how to convert the TCN checkpoint to the PyTorch format, is the parameter saving order something like [Layer0_Weights, Layer0_Bias, Layer1_Weights, Layer1_Bias, ...]? Also, is the memory layout "row-major"? <img width="1100" height="482" alt="Image" src="https://github.com/user-attachments/assets/e6958aa3-ff39-4214-bb9d-23621bb88b78" /> <img width="906" height="514" alt="Image" src="https://github.com/user-attachments/assets/4b32bce6-33e0-45a2-8628-6892a7734fb4" /> <img width="964" height="230" alt="Image" src="https://github.com/user-attachments/assets/39c62cc7-52ce-4dfd-b8bd-f02ecf7e8170" />
关闭于 2025-08-10 3 条评论