Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/XiaLiPKU/EMANet
/ functions
Functions
153 in github.com/XiaLiPKU/EMANet
⨍
Functions
153
◇
Types & classes
39
↓ 11 callers
Method
put
(self, result)
bn_lib/nn/modules/comm.py:26
↓ 10 callers
Method
assertTensorClose
(self, a, b, atol=1e-3, rtol=1e-3)
bn_lib/nn/modules/unittest.py:24
↓ 8 callers
Method
get
(self)
bn_lib/nn/modules/comm.py:32
↓ 6 callers
Function
_find_bn
(module)
bn_lib/nn/modules/tests/test_sync_batchnorm.py:30
↓ 6 callers
Method
_make_layer
(self, block, planes, blocks, stride=1, dilation=1, grids=None)
network.py:99
↓ 5 callers
Method
__init__
(self, c, k, stage_num=3)
network.py:194
↓ 5 callers
Method
_checkBatchNormResult
Check the forward and backward for the customized batch normalization.
bn_lib/nn/modules/tests/test_sync_batchnorm.py:44
↓ 5 callers
Function
_unsqueeze_ft
add new dementions at the front and the tail
bn_lib/nn/modules/batchnorm.py:29
↓ 3 callers
Function
get_params
(model, key)
train.py:22
↓ 3 callers
Method
save_checkpoints
(self, name)
train.py:107
↓ 2 callers
Method
_check_input_dim
(self, input)
bn_lib/nn/modules/batchnorm.py:189
↓ 2 callers
Method
_l2norm
Normlize the inp tensor with l2-norm. Returns a tensor where each sub-tensor of input along the given dim is normalized such that th
network.py:250
↓ 2 callers
Method
_process_next_batch
(self, batch)
bn_lib/utils/data/dataloader.py:299
↓ 2 callers
Method
_put_indices
(self)
bn_lib/utils/data/dataloader.py:290
↓ 2 callers
Method
_shutdown_workers
(self)
bn_lib/utils/data/dataloader.py:314
↓ 2 callers
Function
_sum_ft
sum over the first and last dimention
bn_lib/nn/modules/batchnorm.py:24
↓ 2 callers
Function
as_numpy
(v)
bn_lib/nn/modules/unittest.py:17
↓ 2 callers
Function
async_copy_to
(obj, dev, main_stream=None)
bn_lib/nn/parallel/data_parallel.py:12
↓ 2 callers
Function
cal_scores
(hist)
metric.py:41
↓ 2 callers
Function
ensure_dir
(dir_path)
train.py:39
↓ 2 callers
Function
execute_replication_callbacks
Execute an replication callback `__data_parallel_replicate__` on each module created by original replication. The callback will be invoked w
bn_lib/nn/modules/replicate.py:27
↓ 2 callers
Function
fetch
(image_path, label_path=None)
dataset.py:15
↓ 2 callers
Function
handy_var
(a, unbias=True)
bn_lib/nn/modules/tests/test_numeric_batchnorm.py:18
↓ 2 callers
Function
pin_memory_batch
(batch)
bn_lib/utils/data/dataloader.py:142
↓ 1 callers
Method
__data_parallel_replicate__
(self, ctx, copy_id)
bn_lib/nn/modules/batchnorm.py:83
↓ 1 callers
Method
__init__
(self, data_root=settings.DATA_ROOT, split='val')
dataset.py:127
↓ 1 callers
Method
__init__
(self, dataset, indices)
bn_lib/utils/data/dataset.py:94
↓ 1 callers
Function
_async_copy_stream
(inputs, device_ids)
bn_lib/nn/parallel/data_parallel.py:83
↓ 1 callers
Method
_compute_mean_std
Compute the mean and standard-deviation with sum and square-sum. This method also maintains the moving average on the master device.
bn_lib/nn/modules/batchnorm.py:118
↓ 1 callers
Method
_get_batch
(self)
bn_lib/utils/data/dataloader.py:249
↓ 1 callers
Method
_get_item
(self, idx)
dataset.py:104
↓ 1 callers
Function
_get_stream
Gets a background stream for copying between CPU and GPU
bn_lib/nn/parallel/data_parallel.py:105
↓ 1 callers
Function
_set_SIGCHLD_handler
()
bn_lib/utils/data/dataloader.py:160
↓ 1 callers
Method
_syncParameters
(self, bn1, bn2)
bn_lib/nn/modules/tests/test_sync_batchnorm.py:37
↓ 1 callers
Function
crop
(image, label=None)
dataset.py:68
↓ 1 callers
Method
cumsum
(sequence)
bn_lib/utils/data/dataset.py:61
↓ 1 callers
Function
dict_gather
Gathers variables from different GPUs on a specified device (-1 means the CPU), with dictionary support.
bn_lib/nn/parallel/data_parallel.py:28
↓ 1 callers
Function
fast_hist
(label_true, label_pred)
metric.py:6
↓ 1 callers
Function
flip
(image, label=None)
dataset.py:80
↓ 1 callers
Function
gather_map
(outputs)
bn_lib/nn/parallel/data_parallel.py:33
↓ 1 callers
Method
inf_batch
(self, image, label)
eval.py:46
↓ 1 callers
Method
load_checkpoints
(self, name)
eval.py:34
↓ 1 callers
Method
load_checkpoints
(self, name)
train.py:115
↓ 1 callers
Function
main
(ckp_name='final.pth')
eval.py:56
↓ 1 callers
Function
main
(ckp_name='latest.pth')
train.py:145
↓ 1 callers
Function
pad
(image, label=None)
dataset.py:42
↓ 1 callers
Function
pad_inf
(image, label=None)
dataset.py:55
↓ 1 callers
Function
patch_replication_callback
Monkey-patch an existing `DataParallel` object. Add the replication callback. Useful when you have customized `DataParallel` implementation.
bn_lib/nn/modules/replicate.py:70
↓ 1 callers
Function
poly_lr_scheduler
(opt, init_lr, iter, lr_decay_iter, max_iter, power)
train.py:44
↓ 1 callers
Method
register_slave
Register an slave device. Args: identifier: an identifier, usually is the device id. Returns: a `SlavePipe` obj
bn_lib/nn/modules/comm.py:78
↓ 1 callers
Function
resnet
(n_layers, stride)
network.py:149
↓ 1 callers
Method
run_master
Main entry for the master device in each forward pass. The messages were first collected from each devices (including the master devi
bn_lib/nn/modules/comm.py:96
↓ 1 callers
Method
run_slave
(self, msg)
bn_lib/nn/modules/comm.py:49
↓ 1 callers
Function
scale
(image, label=None)
dataset.py:33
↓ 1 callers
Function
test_dt
()
dataset.py:141
↓ 1 callers
Function
test_net
()
network.py:321
↓ 1 callers
Method
train_batch
(self, image, label)
train.py:128
↓ 1 callers
Method
write
(self, out)
train.py:96
Method
__add__
(self, other)
bn_lib/utils/data/dataset.py:22
Method
__del__
(self)
bn_lib/utils/data/dataloader.py:333
Method
__getitem__
(self, idx)
dataset.py:100
Method
__getitem__
(self, index)
bn_lib/utils/data/dataset.py:16
Method
__getitem__
(self, index)
bn_lib/utils/data/dataset.py:42
Method
__getitem__
(self, idx)
bn_lib/utils/data/dataset.py:78
Method
__getitem__
(self, idx)
bn_lib/utils/data/dataset.py:98
Method
__getstate__
(self)
bn_lib/utils/data/dataloader.py:306
Method
__init__
(self, inplanes, planes, stride=1, dilation=1, downsample=None, previous_dilation=1)
network.py:19
Method
__init__
(self, block, layers, num_classes=1000, stride=8)
network.py:58
Method
__init__
(self, c_in, c_out, kernel_size, stride, padding, dilation)
network.py:171
Method
__init__
(self, n_classes, n_layers)
network.py:268
Method
__init__
(self, weight=None, reduction='none', ignore_index=-1)
network.py:311
Method
__init__
(self, dt_split)
eval.py:21
Method
__init__
(self, dt_split)
train.py:54
Method
__init__
(self, data_root, split)
dataset.py:89
Method
__init__
(self, data_root=settings.DATA_ROOT, split='trainaug')
dataset.py:109
Method
__init__
(self, exc_info)
bn_lib/utils/data/dataloader.py:25
Method
__init__
(self, loader)
bn_lib/utils/data/dataloader.py:188
Method
__init__
(self, dataset, batch_size=1, shuffle=False, sampler=None, batch_sampler=None, num_workers=0,
bn_lib/utils/data/dataloader.py:380
Method
__init__
(self, dataset, num_replicas=None, rank=None)
bn_lib/utils/data/distributed.py:25
Method
__init__
(self, data_tensor, target_tensor)
bn_lib/utils/data/dataset.py:37
Method
__init__
(self, datasets)
bn_lib/utils/data/dataset.py:69
Method
__init__
(self, data_source)
bn_lib/utils/data/sampler.py:12
Method
__init__
(self, data_source)
bn_lib/utils/data/sampler.py:29
Method
__init__
(self, data_source)
bn_lib/utils/data/sampler.py:46
Method
__init__
(self, indices)
bn_lib/utils/data/sampler.py:63
Method
__init__
(self, weights, num_samples, replacement=True)
bn_lib/utils/data/sampler.py:84
Method
__init__
(self, sampler, batch_size, drop_last)
bn_lib/utils/data/sampler.py:112
Method
__init__
(self)
bn_lib/nn/modules/comm.py:21
Method
__init__
Args: master_callback: a callback to be invoked after having collected messages from slave devices.
bn_lib/nn/modules/comm.py:67
Method
__init__
(self, num_features, eps=1e-5, momentum=3e-4, affine=True)
bn_lib/nn/modules/batchnorm.py:39
Method
__iter__
(self)
bn_lib/utils/data/dataloader.py:287
Method
__iter__
(self)
bn_lib/utils/data/dataloader.py:418
Method
__iter__
(self)
bn_lib/utils/data/distributed.py:37
Method
__iter__
(self)
bn_lib/utils/data/sampler.py:15
Method
__iter__
(self)
bn_lib/utils/data/sampler.py:32
Method
__iter__
(self)
bn_lib/utils/data/sampler.py:49
Method
__iter__
(self)
bn_lib/utils/data/sampler.py:66
Method
__iter__
(self)
bn_lib/utils/data/sampler.py:89
Method
__iter__
(self)
bn_lib/utils/data/sampler.py:117
Method
__len__
(self)
dataset.py:97
next →
1–100 of 153, ranked by callers