摘要: 我们在训练时如果使用了BN层和Dropout层,我们需要对model进行标识: model.train():在训练时使用BN层和Dropout层,对模型进行更改。 model.eval():在评价时将BN层和Dropout层冻结,这两个操作不会对模型进行更改。 阅读全文
posted @ 2020-02-27 21:24 谦谦谦谦谦谦谦谦 阅读(1938) 评论(0) 推荐(0) 编辑
摘要: 在写代码时发现我们在定义Model时,有两种定义方法: torch.nn.Conv2d()和torch.nn.functional.conv2d() 那么这两种方法到底有什么区别呢,我们通过下述代码看出差别,先拿torch.nn.Conv2d torch.nn.Conv2d class Conv2d 阅读全文
posted @ 2020-02-27 21:15 谦谦谦谦谦谦谦谦 阅读(1121) 评论(0) 推荐(0) 编辑