训练时间

[456sjd]

你好 我想请问下 我GPU就一张：RTX3090 训练样本是MVTec-2K 种类就一种bottle 为什么训练会两个小时还不见训练完成？请问你的训练大概花费了多久时间？

[cnulab]

请问你使用了哪个模型？您可以在日志文件夹中追踪训练进度。

[456sjd]

python runs/run_patchcore.py --data_root data/MVTec-2K --category bottle --gpus 0,1 这是我的训练命令 我的日志是截止到task_15_weight.pkl 后面代码不见打印训练完成 日志也没有更新了 但是GPU显存也一直没有释放 请问这是怎么回事

[cnulab]

如果一张gpu应该设置--gpus 0
这样会把所有任务分配到第一张GPU上，训练和推理应该不会花费太多时间。

[456sjd]

好的 我之前应该是没有设置gpus这个参数 就只出现了task_0_weight.pkl task_5_weight.pkl task_10_weight.pkl task_15_weight.pkl四个文件 应该是在等待其他任务完成所以一直卡着是吗

[cnulab]

是的，没有找到对应的GPU所以一直卡着

[456sjd]

好的 明白了 谢谢您