search

mhamilton723 / STEGO

Unsupervised Semantic Segmentation by Distilling Feature Correspondences
MIT License
711 stars 142 forks source link

enviroment error about pytorch lighting #69

Closed HalvesChen closed 1 year ago

HalvesChen commented 1 year ago

I use pytorch 1.7 and pytorch lighting 0.8.5, but there is a problem that TypeError: __init__() got an unexpected keyword argument 'every_n_train_steps'. I want to know which lighting is available？

KennyChen880127 commented 1 year ago

I'm success to use pytorch-lightning==1.5.10, maybe you can try it!

HalvesChen commented 1 year ago

@KennyChen880127 Thanks！