size_average和reduce参数将被弃用,请改用reduction='sum'。
此报错为pytorch不同版本进行更新迭代时引起的警告,某些参数被取代了
criterion = torch.nn.BCELoss(size_average=False)
改为
criterion = torch.nn.BCELoss(reduction='sum' )
size_average和reduce参数将被弃用,请改用reduction='sum'。
此报错为pytorch不同版本进行更新迭代时引起的警告,某些参数被取代了
criterion = torch.nn.BCELoss(size_average=False)
改为
criterion = torch.nn.BCELoss(reduction='sum' )