Function_distributed_worker(
local_rank,
main_func,
world_size,
num_gpus_per_machine,
machine_rank,
backend,
unicorn/core/launch.py:101
Functionadjust_labels_sz inp_lbs: PyTorch tensor (F, K, H, W) --> (F, K, H*s, W*s) dh: destination h, dw: destination w
external/lib/utils/merge.py:24
Functionadjust_labels_sz inp_lbs: PyTorch tensor (F, K, H, W) --> (F, K, H*s, W*s) dh: destination h, dw: destination w
unicorn/utils/merge.py:24