search

moboehle / B-cos

B-cos Networks: Alignment is All we Need for Interpretability
Other
104 stars 10 forks source link

Where is train.py? #2

Closed JuChanSeo closed 2 years ago

JuChanSeo commented 2 years ago

Hi, Thanks for your interesting paper. I'm trying to train your model but only I can find is Trainer class in training folder and not the training loop code. Can you please add your code for training loop code OR give some guidelines for reproducing your implementation?

Thanks a lot.

nps1ngh commented 2 years ago

Hey,

check out the README.md file here: https://github.com/moboehle/B-cos/tree/main/experiments

JuChanSeo commented 2 years ago

Hey,

check out the README.md file here: https://github.com/moboehle/B-cos/tree/main/experiments

Thanks a lot. It seems like I didn't check carefully