2024年12月31日
model
摘要： 所有的网络都要继承nn.Module，这个类中封装好了很多函数，例如train(),to, cuda, state_dict, load_state_dict等； train model.train() 并不是真的执行训练， 而是PyTorch 中 nn.Module 的一个方法，用于将模型设置为训 阅读全文
posted @ 2024-12-31 08:59 2426051714 阅读(13) 评论(0) 推荐(0)