search

logix-project / logix

AI Logging for Interpretability and Explainability🔬
Apache License 2.0
86 stars 6 forks source link

Conv2d lora #32

Closed hage1005 closed 11 months ago

hage1005 commented 11 months ago

Implement Conv2d Lora and cifar example

Note: I added src_log.view and tgt_log.view (shape[0], shape[1], -1) because conv2d weight layer shape has more than 2 dimensions.

sangkeun00 commented 11 months ago

LGTM!