Dan Alistarh

bug in prune-weights-ResNet18.py?

1

Hi, I was looking into the ResNet18 pytorch code, and noticed the following sequence: ```# Specify layer type, C for convolution, F for fully-connected if layer_name == 'fc': layer_type =...

Error when running baseline inference experiments

2

Hi, When running the first inference experiment in the README > python run.py experiment=imagenet-t2tvit-eval.yaml model/t2tattn_cfg=full datamodule.data_dir=/home/dan/ILSVRC/ eval.ckpt=checkpoints/t2tvit/81.7_T2T_ViTt_14.pth.tar I get the following error: ```Traceback (most recent call last): File "run.py", line...

Transformer & LSTM setup

1

Hi, Thank you for the really nice paper and for the code release. Would it also be possible to release the Transformer and LSTM setups? Thanks, Dan