MCPcopy Create free account

hub / github.com/dailenson/SDT / functions

Functions101 in github.com/dailenson/SDT

↓ 6 callersMethodwith_pos_embed
(self, tensor, pos: Optional[Tensor])
models/transformer.py:206
↓ 5 callersMethod__init__
(self, d_model=512, nhead=8, num_encoder_layers=6, num_decoder_layers=6, dim_feedforward=2048
models/transformer.py:18
↓ 4 callersFunctioncoords_render
(coordinates, split, width, height, thickness, board=5)
utils/util.py:48
↓ 3 callersFunctionassert_and_infer_cfg
Call this function in your script after you have finished setting all cfg values that are necessary (e.g., merging a config from a file, merging
parse_config.py:186
↓ 3 callersFunctioncorrds2xys
(coordinates)
utils/util.py:165
↓ 3 callersMethodinference
(self, style_imgs, char_img, max_len)
models/model.py:151
↓ 2 callersMethod__init__
(self, hid_dim, dropout=0.1)
models/model.py:196
↓ 2 callersFunction_check_and_coerce_cfg_value_type
Checks that `value_a`, which is intended to replace `value_b` is of the right type. The type is correct if it matches exactly or is one of a few
parse_config.py:318
↓ 2 callersFunction_decode_cfg_value
Decodes a raw config value (e.g., from a yaml config files or command line argument) into a Python object.
parse_config.py:285
↓ 2 callersFunction_get_activation_fn
Return an activation function given a string
models/transformer.py:286
↓ 2 callersFunction_get_clones
(module, N)
models/transformer.py:269
↓ 2 callersFunction_merge_a_into_b
Merge config dictionary a into config dictionary b, clobbering the options in b whenever they are also specified in a.
parse_config.py:252
↓ 2 callersFunctiondxdynp_to_list
(coordinates)
utils/util.py:135
↓ 2 callersFunctiongenerate_square_subsequent_mask
(sz: int)
models/model.py:230
↓ 2 callersFunctionget_mixture_coef
(output)
models/gmm.py:4
↓ 2 callersFunctionlength_to_mask
length: B. return B x max_len. If max_len is None, then max of length will be used.
models/eval_model.py:99
↓ 2 callersMethodtrain
start training iterations
trainer/trainer.py:102
↓ 2 callersMethodwith_pos_embed
(self, tensor, pos: Optional[Tensor])
models/transformer.py:143
↓ 1 callersMethod__init__
(self, num_class=240, vote=False)
models/eval_model.py:8
↓ 1 callersMethod__init__
(self, d_model=256, nhead=8, num_encoder_layers=3, dim_feedforward=2048, dropout=0.1, activat
models/encoder.py:9
↓ 1 callersMethod_progress
(self, step, loss, time_left)
trainer/trainer.py:124
↓ 1 callersMethod_reset_parameters
(self)
models/transformer.py:40
↓ 1 callersMethod_reset_parameters
(self)
models/encoder.py:39
↓ 1 callersMethod_reset_parameters
(self)
models/model.py:53
↓ 1 callersMethod_save_checkpoint
(self, step)
trainer/trainer.py:130
↓ 1 callersMethod_train_iter
(self, data, step)
trainer/trainer.py:27
↓ 1 callersMethod_valid_iter
(self, step)
trainer/trainer.py:76
↓ 1 callersMethod_vis_genarate_samples
(self, gt_coords, preds, character_id, step)
trainer/trainer.py:135
↓ 1 callersFunctionfast_norm_len_dtw
start test iterations
utils/metrics.py:9
↓ 1 callersFunctionfix_seed
(random_seed)
utils/util.py:100
↓ 1 callersMethodforward_post
(self, src, src_mask: Optional[Tensor] = None,
models/transformer.py:146
↓ 1 callersMethodforward_post
(self, tgt, memory, tgt_mask: Optional[Tensor] = None, memory_mask:
models/transformer.py:209
↓ 1 callersMethodforward_pre
(self, src, src_mask: Optional[Tensor] = None, src_key_padding_mask: O
models/transformer.py:161
↓ 1 callersMethodforward_pre
(self, tgt, memory, tgt_mask: Optional[Tensor] = None, memory_mask: Op
models/transformer.py:232
↓ 1 callersFunctionget_content_score
set model, criterion and optimizer
utils/metrics.py:59
↓ 1 callersFunctionget_seq_from_gmm
(gmm_pred)
models/gmm.py:22
↓ 1 callersFunctionget_style_score
(test_loader,pretrained_model)
utils/metrics.py:36
↓ 1 callersMethodimmutable
Set immutability to is_immutable and recursively apply the setting to all nested AttrDicts.
parse_config.py:26
↓ 1 callersFunctionload_specific_dict
(model, pretrained_model, par)
utils/util.py:111
↓ 1 callersFunctionmain
load config file into cfg
train.py:11
↓ 1 callersFunctionmain
load config file into cfg
test.py:12
↓ 1 callersFunctionmain
load config file into cfg
user_generate.py:12
↓ 1 callersFunctionmain
(opt)
evaluate.py:9
↓ 1 callersFunctionmask_avr_pooling
(x, l)
models/eval_model.py:91
↓ 1 callersFunctionnormalize_xys
(xys)
utils/util.py:11
↓ 1 callersMethodrandom_double_sampling
Sample the positive pair (i.e., o and o^+) within a character by per-sample shuffling. Per-sample shuffling is done by argsort random
models/model.py:58
↓ 1 callersFunctionset_log
(output_dir, cfg_file, log_name)
utils/logger.py:6
↓ 1 callersFunctiontf_2d_normal
(x1, x2, mu1, mu2, s1, s2, rho)
models/loss.py:114
↓ 1 callersFunctionwriteCache
(env, cache)
utils/util.py:126
Method__getitem__
(self, index)
data_loader/loader.py:73
Method__getitem__
(self, index)
data_loader/loader.py:164
Method__getitem__
(self, index)
data_loader/loader.py:214
Method__getitem__
(self, index)
data_loader/loader.py:264
Method__getitem__
(self, index)
data_loader/loader.py:321
Method__init__
(self, *args)
parse_config.py:23
Method__init__
(self, root='data', dataset='CHINESE', is_train=True, num_img = 15)
data_loader/loader.py:26
Method__init__
(self, data_path)
data_loader/loader.py:150
Method__init__
(self, root='data', dataset='CHINESE', style_path='style_samples')
data_loader/loader.py:205
Method__init__
(self, root=None, is_train=True, num_img=15)
data_loader/loader.py:235
Method__init__
(self, data_path='lmdb', is_train=True)
data_loader/loader.py:289
Method__init__
(self, model, criterion, optimizer, data_loader, logs, char_dict, valid_data_loader=None)
trainer/trainer.py:13
Method__init__
(self, temperature=0.07, contrast_mode='all', base_temperature=0.07)
models/loss.py:9
Method__init__
(self, nclass=3755)
models/eval_model.py:46
Method__init__
(self, encoder_layer, num_layers, norm=None)
models/transformer.py:61
Method__init__
(self, decoder_layer, num_layers, norm=None, return_intermediate=False)
models/transformer.py:85
Method__init__
(self, d_model, nhead, dim_feedforward=2048, dropout=0.1, activation="relu", normalize_before
models/transformer.py:126
Method__init__
(self, d_model, nhead, dim_feedforward=2048, dropout=0.1, activation="relu", normalize_before
models/transformer.py:186
Method__init__
(self, dropout, dim, max_len=500)
models/transformer.py:307
Method__init__
(self, d_model=512, num_encoder_layers=3, num_classes=6763)
models/encoder.py:33
Method__init__
(self, d_model=512, nhead=8, num_encoder_layers=2, num_head_layers= 1, wri_dec_layers=2, gly_
models/model.py:14
Method__init__
(self, hid_dim, dropout=0.1)
models/model.py:211
Method__len__
(self)
data_loader/loader.py:114
Method__len__
(self)
data_loader/loader.py:179
Method__len__
(self)
data_loader/loader.py:211
Method__len__
(self)
data_loader/loader.py:282
Method__len__
(self)
data_loader/loader.py:348
Function_rebuild_tensor_v2
(storage, storage_offset, size, stride, requires_grad, backward_hooks)
parse_config.py:202
Functionbuild_transformer
(args)
models/transformer.py:273
Methodcollate_fn_
(self, batch_data)
data_loader/loader.py:117
Methodcollate_fn_
(self, batch_data)
data_loader/loader.py:182
Methodcollate_fn_
(self, batch_data)
data_loader/loader.py:351
Methodforward
Compute loss for model. If both `labels` and `mask` are None, it degenerates to SimCLR unsupervised loss: https://arxiv.org/pdf/2002.0
models/loss.py:16
Methodforward
(self,x)
models/eval_model.py:18
Methodforward
(self, x, l)
models/eval_model.py:69
Methodforward
(self, src, query_embed, y_ind)
models/transformer.py:45
Methodforward
(self, src, mask: Optional[Tensor] = None, src_key_padding_mask: Optional[Tens
models/transformer.py:67
Methodforward
(self, tgt, memory, tgt_mask: Optional[Tensor] = None, memory_mask: Optional[T
models/transformer.py:92
Methodforward
(self, src, src_mask: Optional[Tensor] = None, src_key_padding_mask: Optional[
models/transformer.py:175
Methodforward
(self, tgt, memory, tgt_mask: Optional[Tensor] = None, memory_mask: Optional[T
models/transformer.py:255
Methodforward
Embed inputs. Args: emb (FloatTensor): Sequence of word vectors ``(seq_len, batch_size, self.dim)`` s
models/transformer.py:323
Methodforward
(self, x)
models/encoder.py:20
Methodforward
(self, x)
models/encoder.py:44
Methodforward
(self, style_imgs, seq, char_img)
models/model.py:80
Methodforward
(self, seq)
models/model.py:202
Methodforward
(self, seq)
models/model.py:217
Functionget_pen_loss
(z_pi, z_mu1, z_mu2, z_sigma1, z_sigma2, z_corr, z_pen_logits, x1_data, x2_data, pen_data)
models/loss.py:96
Methodis_immutable
(self)
parse_config.py:39
Functionmask_avr_pooling_rnn
(x, l)
models/eval_model.py:83
Functionmerge_cfg_from_cfg
Merge `cfg_other` into the global config.
parse_config.py:221
Functionmerge_cfg_from_file
Load a yaml config file and merge it into the global config.
parse_config.py:212
next →1–100 of 101, ranked by callers