목록Code Study (1)

soyeonland

Learning both weights and connections for Efficient Neural Networks-(1)

learning rate 확인하고 싶을 때 def get_lr(optimizer): for param_group in optimizer.param_groups: return param_group['lr'] model.children() returns model.children() is a generator that returns layers of the model from which you can extract your parameter tensors using .weight or .bias[1] print(list(model.children())) [Conv2d(1, 6, kernel_size=(3, 3), stride=(1, 1)), Conv2d(6, 16, kernel_size=(3, 3), str..
Study/Code Review 2020. 3. 22. 20:05
이전 Prev 1 Next 다음