MCPcopy Create free account

hub / github.com/LTH14/JiT / functions

Functions82 in github.com/LTH14/JiT

⨍ Functions 82◇ Types & classes 14
↓ 30 callersFunctionprint
(*args, **kwargs)
util/misc.py:165
↓ 7 callersMethod__init__
( self, input_size=256, patch_size=16, in_channels=3, hidden_size=1024
model_jit.py:209
↓ 3 callersMethod_forward_sample
(self, z, t, labels)
denoiser.py:91
↓ 3 callersFunctionis_dist_avail_and_initialized
()
util/misc.py:176
↓ 3 callersFunctionmodulate
(x, shift, scale)
model_jit.py:13
↓ 2 callersMethod__init__
LlamaRMSNorm is equivalent to T5LayerNorm
util/model_util.py:138
↓ 2 callersFunctionbroadcat
(tensors, dim = -1)
util/model_util.py:15
↓ 2 callersFunctioncenter_crop_arr
Center cropping implementation from ADM. https://github.com/openai/guided-diffusion/blob/8fb3ad9197f16bbc40620447b2742e13458d2831/guided_diff
util/crop.py:5
↓ 2 callersFunctionevaluate
(model_without_ddp, args, epoch, batch_size=64, log_writer=None)
engine_jit.py:67
↓ 2 callersFunctionget_1d_sincos_pos_embed_from_grid
embed_dim: output dimension for each position pos: a list of positions to be encoded: size (M,) out: (M, D)
util/model_util.py:183
↓ 2 callersFunctionget_world_size
()
util/misc.py:184
↓ 2 callersFunctionrotate_half
(x)
util/model_util.py:32
↓ 2 callersFunctionsetup_for_distributed
This function disables printing when not in master process
util/misc.py:159
↓ 2 callersMethodupdate
(self, value, n=1)
util/misc.py:26
↓ 2 callersMethodupdate
(self, **kwargs)
util/misc.py:80
↓ 1 callersMethod_euler_step
(self, z, t, t_next, labels)
denoiser.py:108
↓ 1 callersMethodadd_meter
(self, name, meter)
util/misc.py:109
↓ 1 callersMethoddrop_labels
(self, labels)
denoiser.py:40
↓ 1 callersMethodgenerate
(self, labels)
denoiser.py:68
↓ 1 callersFunctionget_2d_sincos_pos_embed
grid_size: int of the grid height and width return: pos_embed: [grid_size*grid_size, embed_dim] or [1+grid_size*grid_size, embed_dim] (w/
util/model_util.py:154
↓ 1 callersFunctionget_2d_sincos_pos_embed_from_grid
(embed_dim, grid)
util/model_util.py:172
↓ 1 callersFunctionget_args_parser
()
main_jit.py:23
↓ 1 callersFunctionget_rank
()
util/misc.py:190
↓ 1 callersMethodinitialize_weights
(self)
model_jit.py:279
↓ 1 callersFunctionis_main_process
()
util/misc.py:196
↓ 1 callersMethodlog_every
(self, iterable, print_freq, header=None)
util/misc.py:112
↓ 1 callersFunctionmain
(args)
main_jit.py:115
↓ 1 callersFunctionmain
()
prepare_ref.py:8
↓ 1 callersMethodsample_t
(self, n: int, device=None)
denoiser.py:45
↓ 1 callersFunctionsave_on_master
(*args, **kwargs)
util/misc.py:200
↓ 1 callersFunctionscaled_dot_product_attention
(query, key, value, dropout_p=0.0)
model_jit.py:94
↓ 1 callersMethodtimestep_embedding
Create sinusoidal timestep embeddings. :param t: a 1-D Tensor of N indices, one per batch element. These ma
model_jit.py:54
↓ 1 callersFunctiontrain_one_epoch
(model, model_without_ddp, data_loader, optimizer, device, epoch, log_writer=None, args=None)
engine_jit.py:16
↓ 1 callersMethodunpatchify
x: (N, T, patch_size**2 * C) imgs: (N, H, W, C)
model_jit.py:317
↓ 1 callersMethodupdate_ema
(self)
denoiser.py:125
FunctionJiT_B_16
(**kwargs)
model_jit.py:362
FunctionJiT_B_32
(**kwargs)
model_jit.py:366
FunctionJiT_H_16
(**kwargs)
model_jit.py:378
FunctionJiT_H_32
(**kwargs)
model_jit.py:382
FunctionJiT_L_16
(**kwargs)
model_jit.py:370
FunctionJiT_L_32
(**kwargs)
model_jit.py:374
Method__getattr__
(self, attr)
util/misc.py:89
Method__init__
(self, img_size=224, patch_size=16, in_chans=3, pca_dim=768, embed_dim=768, bias=True)
model_jit.py:20
Method__init__
(self, hidden_size, frequency_embedding_size=256)
model_jit.py:44
Method__init__
(self, num_classes, hidden_size)
model_jit.py:84
Method__init__
(self, dim, num_heads=8, qkv_bias=True, qk_norm=True, attn_drop=0., proj_drop=0.)
model_jit.py:108
Method__init__
( self, dim: int, hidden_dim: int, drop=0.0, bias=True )
model_jit.py:142
Method__init__
(self, hidden_size, patch_size, out_channels)
model_jit.py:166
Method__init__
(self, hidden_size, num_heads, mlp_ratio=4.0, attn_drop=0.0, proj_drop=0.0)
model_jit.py:184
Method__init__
( self, args )
denoiser.py:7
Method__init__
(self, window_size=20, fmt=None)
util/misc.py:18
Method__init__
(self, delimiter="\t")
util/misc.py:76
Method__init__
( self, dim, pt_seq_len, ft_seq_len=None, custom_freqs = None,
util/model_util.py:40
Method__init__
( self, dim, pt_seq_len=16, ft_seq_len=None, custom_freqs = None,
util/model_util.py:87
Method__str__
(self)
util/misc.py:66
Method__str__
(self)
util/misc.py:97
Method_basic_init
(module)
model_jit.py:281
Method_heun_step
(self, z, t, t_next, labels)
denoiser.py:114
Functionadd_weight_decay
(model, weight_decay=0, skip_list=())
util/misc.py:240
Functionadjust_learning_rate
Decay the learning rate with half-cycle cosine after warmup
util/lr_sched.py:4
Functionall_reduce_mean
(x)
util/misc.py:281
Methodavg
(self)
util/misc.py:50
Methodforward
(self, x)
model_jit.py:32
Methodforward
(self, t)
model_jit.py:74
Methodforward
(self, labels)
model_jit.py:89
Methodforward
(self, x, rope)
model_jit.py:121
Methodforward
(self, x)
model_jit.py:155
Methodforward
(self, x, c)
model_jit.py:176
Methodforward
(self, x, c, feat_rope=None)
model_jit.py:198
Methodforward
x: (N, C, H, W) t: (N,) y: (N,)
model_jit.py:331
Methodforward
(self, x, labels)
denoiser.py:49
Methodforward
(self, t, start_index = 0)
util/model_util.py:77
Methodforward
(self, t)
util/model_util.py:134
Methodforward
(self, hidden_states)
util/model_util.py:146
Methodglobal_avg
(self)
util/misc.py:55
Functioninit_distributed_mode
(args)
util/misc.py:205
Methodmax
(self)
util/misc.py:59
Methodmedian
(self)
util/misc.py:45
Functionsave_model
(args, model_without_ddp, optimizer, epoch, epoch_name=None)
util/misc.py:255
Methodsynchronize_between_processes
Warning: does not synchronize the deque!
util/misc.py:31
Methodsynchronize_between_processes
(self)
util/misc.py:105
Methodvalue
(self)
util/misc.py:63