The onnxvitssharp from gaomengkai

onnxvitssharp's People

Contributors

Stargazers

Watchers

onnxvitssharp's Issues

关于Onnx模型格式以及文件不匹配的提问

如题，我昨天通过来自此链接的训练Notebook得到我预期的模型，可是最后的文件结构是这样的：

total 1926384
-rw-r--r-- 1 root root      2170 Nov 29 08:44 config.json
-rw-r--r-- 1 root root 187026900 Nov 29 12:33 D_18100.pth
-rw-r--r-- 1 root root 187026900 Nov 29 12:35 D_18200.pth
-rw-r--r-- 1 root root 187026900 Nov 29 12:36 D_18300.pth
-rw-r--r-- 1 root root 187026900 Nov 29 12:38 D_18400.pth
-rw-r--r-- 1 root root 187027527 Nov 29 12:38 D_latest.pth
drwxr-xr-x 2 root root      4096 Nov 29 08:44 eval
-rw-r--r-- 1 root root        88 Nov 28 23:38 events.out.tfevents.1701185938.ElinsLaptop.9330.0
-rw-r--r-- 1 root root        88 Nov 28 23:54 events.out.tfevents.1701186848.ElinsLaptop.14691.0
-rw-r--r-- 1 root root        88 Nov 29 00:26 events.out.tfevents.1701188780.ElinsLaptop.12482.0
-rw-r--r-- 1 root root        88 Nov 29 00:27 events.out.tfevents.1701188846.ElinsLaptop.12995.0
-rw-r--r-- 1 root root        88 Nov 29 00:35 events.out.tfevents.1701189323.ElinsLaptop.17637.0
-rw-r--r-- 1 root root        88 Nov 29 00:37 events.out.tfevents.1701189428.ElinsLaptop.18643.0
-rw-r--r-- 1 root root        88 Nov 29 00:39 events.out.tfevents.1701189573.ElinsLaptop.466.0
-rw-r--r-- 1 root root   4084328 Nov 29 00:48 events.out.tfevents.1701189877.ElinsLaptop.878.0
-rw-r--r-- 1 root root 238198058 Nov 29 12:38 events.out.tfevents.1701218676.ElinsLaptop.701.0
-rw-r--r-- 1 root root 158898430 Nov 29 12:33 G_18100.pth
-rw-r--r-- 1 root root 158898430 Nov 29 12:35 G_18200.pth
-rw-r--r-- 1 root root 158898430 Nov 29 12:36 G_18300.pth
-rw-r--r-- 1 root root 158898430 Nov 29 12:38 G_18400.pth
-rw-r--r-- 1 root root        40 Nov 28 23:38 githash
-rw-r--r-- 1 root root 158899868 Nov 29 12:38 G_latest.pth
-rw-r--r-- 1 root root    588432 Nov 29 12:38 train.log

所以请问针对，我的情况，想使用您的代码库需要如何打包我的模型呢？

Recommend Projects