Haoli Bai

Hi! This is a really interesting work, and thanks for sharing the code! May I know how to perform searching and training with this fast splitting method? BTW, do we...

Where to download the image

2

Thanks for sharing this data. Do you know where to download the images of screen2words?

Heavy cpu usage for the backward pass in method 1 (FunLSQ)

1

In my case, the following code seems to consume heavy cpu usage during the backward pass in FunLSQ. (>1000% cpu usage). ``` indicate_middle = torch.ones(indicate_small.shape).to(indicate_small.device) - indicate_small - indicate_big ```...

如题

Hi, I am trying to add some new learnable parameters inside ColumnParallelLinear/RowParallelLinear, and the following is an example code snippet: ```python class ColumnParallelLinear(torch.nn.Module): """Linear layer with column parallelism. The linear...