LI JING

The code only works on num_queries=1

2

https://github.com/megvii-research/DPGN/blob/b9401115b43319544c07718890e115e7f4fc2a1d/main.py#L316 `self.pred_loss(query_node_pred_generation, query_label.long()).mean()` For example, 5 way 1 shot, num_queries=1, `query_node_pred_generation` has shape of [batch_size, 5, 5] `query_label` has shape of [batch_size, 5] 5 way 1 shot, num_queries=2, `query_node_pred_generation` have...

Bug on resume training on single GPU and multiple GPUs

2

### System Info ```Shell accelerate: 0.12.0 ``` ### Information - [X] The official example scripts - [ ] My own modified scripts ### Tasks - [X] One of the scripts...