Commit Graph

4 Commits

Author SHA1 Message Date
bba39adcfc 优化训练过程 2025-03-20 16:41:51 +08:00
358bb40a2a 修改一些依赖的版本 2025-03-18 15:23:22 +08:00
ebc4e9df4f 使用cuda版torch进行训练和推理 2025-03-17 15:45:46 +08:00
ff7b6fda0e 增加torch cuda版本 2025-03-17 10:39:00 +08:00