Open jabader97 opened 2 years ago
Added test case for convolutional layer (when equal to linear layer) and commenting in KFAC methods to compare with KFAC paper.
Added test case for convolutional layer (when equal to linear layer) and commenting in KFAC methods to compare with KFAC paper.