MCPcopy Create free account

hub / github.com/FreeformRobotics/OTS / functions

Functions180 in github.com/FreeformRobotics/OTS

⨍ Functions 180◇ Types & classes 46
↓ 11 callersMethodput
(self, result)
lib/nn/modules/comm.py:26
↓ 10 callersMethodassertTensorClose
(self, a, b, atol=1e-3, rtol=1e-3)
lib/nn/modules/unittest.py:24
↓ 8 callersMethodget
(self)
lib/nn/modules/comm.py:32
↓ 8 callersMethodround2nearest_multiple
(self, x, p)
dataset.py:66
↓ 6 callersFunction_find_bn
(module)
lib/nn/modules/tests/test_sync_batchnorm.py:30
↓ 5 callersMethod_checkBatchNormResult
Check the forward and backward for the customized batch normalization.
lib/nn/modules/tests/test_sync_batchnorm.py:44
↓ 5 callersFunction_unsqueeze_ft
add new dementions at the front and the tail
lib/nn/modules/batchnorm.py:29
↓ 5 callersFunctionconv3x3
3x3 convolution with padding
models/resnet.py:18
↓ 5 callersFunctionimresize
(im, size, interp='bilinear')
dataset.py:9
↓ 4 callersMethod_make_layer
(self, block, planes, blocks, stride=1)
models/resnet.py:126
↓ 3 callersMethod__init__
(self, num_class=150, fc_dim=4096, use_softmax=False, pool_scales=(1, 2, 3, 6))
models/models.py:176
↓ 3 callersMethod_add_weighted
return *dest* by `dest := dest*alpha + delta*beta + bias`
lib/nn/modules/batchnorm.py:119
↓ 3 callersMethodimg_transform
(self, img)
dataset.py:53
↓ 3 callersFunctionload_url
(url, model_dir='./pretrained', map_location=None)
models/utils.py:10
↓ 2 callersMethod__init__
(self, root_dataset, odgt, opt, **kwargs)
dataset.py:207
↓ 2 callersMethod__init__
(self, in_dim, one_hot_cls_num)
models/ots.py:42
↓ 2 callersMethod__init__
(self, block, layers, num_classes=1000)
models/resnet.py:97
↓ 2 callersMethod_check_input_dim
(self, input)
lib/nn/modules/batchnorm.py:198
↓ 2 callersMethod_process_next_batch
(self, batch)
lib/utils/data/dataloader.py:302
↓ 2 callersMethod_put_indices
(self)
lib/utils/data/dataloader.py:293
↓ 2 callersMethod_shutdown_workers
(self)
lib/utils/data/dataloader.py:317
↓ 2 callersFunction_sum_ft
sum over the first and last dimention
lib/nn/modules/batchnorm.py:24
↓ 2 callersFunctionas_numpy
(obj)
lib/utils/th.py:18
↓ 2 callersFunctionas_numpy
(v)
lib/nn/modules/unittest.py:17
↓ 2 callersFunctionasync_copy_to
(obj, dev, main_stream=None)
lib/nn/parallel/data_parallel.py:13
↓ 2 callersMethodcumsum
(sequence)
lib/utils/data/dataset.py:61
↓ 2 callersFunctionexecute_replication_callbacks
Execute an replication callback `__data_parallel_replicate__` on each module created by original replication. The callback will be invoked w
lib/nn/modules/replicate.py:27
↓ 2 callersFunctionhandy_var
(a, unbias=True)
lib/nn/modules/tests/test_numeric_batchnorm.py:18
↓ 2 callersFunctionpin_memory_batch
(batch)
lib/utils/data/dataloader.py:145
↓ 2 callersMethodsegm_transform
(self, segm)
dataset.py:60
↓ 2 callersFunctionweight_init
(module: nn.Module)
models/ots.py:9
↓ 1 callersMethod__data_parallel_replicate__
(self, ctx, copy_id)
lib/nn/modules/batchnorm.py:88
↓ 1 callersMethod__init__
(self, dataset, indices)
lib/utils/data/dataset.py:94
↓ 1 callersFunction_async_copy_stream
(inputs, device_ids)
lib/nn/parallel/data_parallel.py:82
↓ 1 callersMethod_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.
lib/nn/modules/batchnorm.py:123
↓ 1 callersMethod_get_batch
(self)
lib/utils/data/dataloader.py:252
↓ 1 callersFunction_get_stream
Gets a background stream for copying between CPU and GPU
lib/nn/parallel/data_parallel.py:104
↓ 1 callersMethod_get_sub_batch
(self)
dataset.py:85
↓ 1 callersFunction_set_SIGCHLD_handler
()
lib/utils/data/dataloader.py:163
↓ 1 callersMethod_syncParameters
(self, bn1, bn2)
lib/nn/modules/tests/test_sync_batchnorm.py:37
↓ 1 callersMethodadd
(self, val, weight)
utils.py:55
↓ 1 callersMethodbuild_decoder
(arch='ppm', fc_dim=512, num_class=150, weights='', use_softmax=Fa
models/models.py:81
↓ 1 callersMethodbuild_encoder
(arch='resnet50dilated', fc_dim=512, weights='')
models/models.py:62
↓ 1 callersFunctionclassify_step
(logit, classes)
functions.py:156
↓ 1 callersFunctiondefault_argument_parser
()
functions.py:24
↓ 1 callersFunctiondict_gather
Gathers variables from different GPUs on a specified device (-1 means the CPU), with dictionary support.
lib/nn/parallel/data_parallel.py:27
↓ 1 callersFunctiongather_map
(outputs)
lib/nn/parallel/data_parallel.py:32
↓ 1 callersFunctionget_img_list
(path)
functions.py:82
↓ 1 callersFunctionget_obj_onehot_vector
(img_name, one_hot)
functions.py:149
↓ 1 callersMethodget_object_feature
(self, segSize, img_resized_list, batch_data)
main.py:34
↓ 1 callersFunctionimg2onehot
(img_name, one_hot)
functions.py:143
↓ 1 callersMethodinitialize
(self, val, weight)
utils.py:42
↓ 1 callersFunctionload_checkpoint
(checkpoint, obj_model, classifier)
functions.py:134
↓ 1 callersFunctionload_classes
(class_file)
functions.py:118
↓ 1 callersFunctionload_dict
(filename)
functions.py:72
↓ 1 callersFunctionload_seg_module
(cfg)
functions.py:88
↓ 1 callersFunctionload_test_data
(cfg)
functions.py:104
↓ 1 callersMethodparse_input_list
(self, odgt, max_sample=-1, start_idx=-1, end_idx=-1)
dataset.py:38
↓ 1 callersMethodpixel_acc
(self, pred, label)
models/models.py:12
↓ 1 callersMethodregister_slave
Register an slave device. Args: identifier: an identifier, usually is the device id. Returns: a `SlavePipe` obj
lib/nn/modules/comm.py:78
↓ 1 callersMethodrun_master
Main entry for the master device in each forward pass. The messages were first collected from each devices (including the master devi
lib/nn/modules/comm.py:96
↓ 1 callersMethodrun_slave
(self, msg)
lib/nn/modules/comm.py:49
↓ 1 callersMethodtest
(self)
main.py:108
↓ 1 callersFunctionunique
(ar, return_index=False, return_inverse=False, return_counts=False)
utils.py:68
↓ 1 callersMethodupdate
(self, val, weight=1)
utils.py:49
Method__add__
(self, other)
lib/utils/data/dataset.py:22
Method__del__
(self)
lib/utils/data/dataloader.py:336
Method__getitem__
(self, index)
dataset.py:110
Method__getitem__
(self, index)
dataset.py:211
Method__getitem__
(self, index)
dataset.py:262
Method__getitem__
(self, index)
lib/utils/data/dataset.py:16
Method__getitem__
(self, index)
lib/utils/data/dataset.py:42
Method__getitem__
(self, idx)
lib/utils/data/dataset.py:78
Method__getitem__
(self, idx)
lib/utils/data/dataset.py:98
Method__getstate__
(self)
lib/utils/data/dataloader.py:309
Method__init__
(self)
utils.py:35
Method__init__
(self, odgt, opt, **kwargs)
dataset.py:23
Method__init__
(self, root_dataset, odgt, opt, batch_per_gpu=1, **kwargs)
dataset.py:71
Method__init__
(self, odgt, opt, **kwargs)
dataset.py:259
Method__init__
(self, cfg, args)
main.py:18
Method__init__
(self, exc_info)
lib/utils/data/dataloader.py:28
Method__init__
(self, loader)
lib/utils/data/dataloader.py:191
Method__init__
(self, dataset, batch_size=1, shuffle=False, sampler=None, batch_sampler=None, num_workers=0,
lib/utils/data/dataloader.py:383
Method__init__
(self, dataset, num_replicas=None, rank=None)
lib/utils/data/distributed.py:25
Method__init__
(self, data_tensor, target_tensor)
lib/utils/data/dataset.py:37
Method__init__
(self, datasets)
lib/utils/data/dataset.py:69
Method__init__
(self, data_source)
lib/utils/data/sampler.py:12
Method__init__
(self, data_source)
lib/utils/data/sampler.py:29
Method__init__
(self, data_source)
lib/utils/data/sampler.py:46
Method__init__
(self, indices)
lib/utils/data/sampler.py:63
Method__init__
(self, weights, num_samples, replacement=True)
lib/utils/data/sampler.py:84
Method__init__
(self, sampler, batch_size, drop_last)
lib/utils/data/sampler.py:112
Method__init__
(self)
lib/nn/modules/comm.py:21
Method__init__
Args: master_callback: a callback to be invoked after having collected messages from slave devices.
lib/nn/modules/comm.py:67
Method__init__
(self, num_features, eps=1e-5, momentum=0.001, affine=True)
lib/nn/modules/batchnorm.py:39
Method__init__
(self)
models/models.py:9
Method__init__
(self, net_enc, net_dec, crit, deep_sup_scale=None)
models/models.py:22
Method__init__
(self, orig_resnet, dilate_scale=8)
models/models.py:112
Method__init__
(self, in_dim, compress)
models/ots.py:16
Method__init__
(self, num_classes, in_dim)
models/ots.py:71
next →1–100 of 180, ranked by callers