kirill neklyudov

Unfortunately, this project was done long time ago and I've lost all the checkpoints that I've made. You can reproduce results of the pretrained model using [this script](https://github.com/necludov/group-sparsity-sbp/blob/master/scripts/vgglike/vgglike.py).

@levskaya this also works, but then I couldn't use it with a GPU.

I think I've tried this, but it seems that installing jax this way on colab is not compatible with GPU. I haven't looked deeper into it, to be honest. So...