如何使用pytorch动态图属性训练与不同超参数并行的模型？

作者: 枫
发布时间: 2024-12-20 04:21:36 (7天前)
转自：

2 条回复

0#
回复此人
無口君 | 2019-08-31 10-32

<div class =“post-text”itemprop =“text”> <P> DataLoader独立于模型。这是你可以做的一个例子。 </p> <pre class="lang-py prettyprint-override"> <code> import torch import torch.nn as nn class Model(nn.Module): # Model definition #Two models with different hyper-parameters model1 = Model(args) model2 = Model(args) dataloader = torch.utils.data.DataLoader(args) for data in dataloader: # Train model1 for data in dataloader: # Train model2 </code> </pre> </DIV>

编辑

登录后才能参与评论