Weijie Su

icon indicating a website link www.weijiesu.com icon indicating an email [email protected]

icon company Shanghai AI Laboratory icon location China icon location Researcher at Shanghai AI Laboratory.

Results 4 comments of Weijie Su

caffe make fails on Debian 9 VM [CXX src/caffe/layers/beam_search_node_layer.cpp]

> Change (Remove explicit Argument Types): > **make_pair(partial_sum[src], ix + end_of_sequence_)** > to > **make_pair(partial_sum[src], ix + end_of_sequence_)** > and similarly at other places in the file. > > It...

Input type (torch.cuda.FloatTensor) and weight type (torch.cuda.HalfTensor) should be the same

Facing the same error while using 'O2', have you found the solution?@BenQLange I found this error only appeared when I run code with multiple GPUs, it is ok while using...

Input type (torch.cuda.FloatTensor) and weight type (torch.cuda.HalfTensor) should be the same

In my case, the reason is that I used a non-official SyncBatchNorm implementation and it seems apex couldn't deal with it.

The problem of multitasking parallel processing

@Pter61 What config do you run? And what is the full command you ran? And please provide your pytorch version and cuda version also.