ZhangZhihui's Blog  

2024年10月13日

摘要: class StepByStep(): def __init__(self, model, loss_fn, optimizer): self.device = 'cuda' if torch.cuda.is_available() else 'cpu' self.model = model.to( 阅读全文
posted @ 2024-10-13 17:48 ZhangZhihuiAAA 阅读(2) 评论(0) 推荐(0) 编辑
 
摘要: 阅读全文
posted @ 2024-10-13 09:50 ZhangZhihuiAAA 阅读(7) 评论(0) 推荐(0) 编辑