MCPcopy Create free account

hub / github.com/ShanghaiTech-IMPACT/TeethDreamer / functions

Functions848 in github.com/ShanghaiTech-IMPACT/TeethDreamer

Methodconfigure_optimizers
(self)
instant-nsr-pl/systems/base.py:118
Methodconfigure_optimizers
(self)
ldm/models/autoencoder.py:197
Methodconfigure_optimizers
(self)
ldm/models/autoencoder.py:386
Methodconfigure_optimizers
(self)
ldm/models/diffusion/zero123.py:457
Methodconfigure_optimizers
(self)
ldm/models/diffusion/teeth_dreamer.py:594
Methodconfigure_optimizers
(self)
ldm/models/diffusion/sync_dreamer.py:880
Functionconvert_module_to_f16
(x)
ldm/modules/diffusionmodules/openaimodel.py:25
Functionconvert_module_to_f32
(x)
ldm/modules/diffusionmodules/openaimodel.py:28
Methodconvert_to_fp16
Convert the torso of the model to float16.
ldm/modules/diffusionmodules/openaimodel.py:730
Methodconvert_to_fp16
Convert the torso of the model to float16.
ldm/modules/diffusionmodules/openaimodel.py:963
Methodconvert_to_fp32
Convert the torso of the model to float32.
ldm/modules/diffusionmodules/openaimodel.py:741
Methodconvert_to_fp32
Convert the torso of the model to float32.
ldm/modules/diffusionmodules/openaimodel.py:970
Methodcount_flops
(model, _x, y)
ldm/modules/diffusionmodules/openaimodel.py:376
Methodcount_flops
(model, _x, y)
ldm/modules/diffusionmodules/openaimodel.py:410
Functioncreate_carvekit_interface
()
ldm/util.py:53
Methoddecode
(self, z)
ldm/modules/encoders/modules.py:532
Methoddecode
(self, h, force_not_quantize=False)
ldm/models/autoencoder.py:274
Methoddecode
(self, x, *args, **kwargs)
ldm/models/autoencoder.py:434
Methoddecode_code
(self, code_b)
ldm/models/autoencoder.py:112
Functiondecorator
(cls)
instant-nsr-pl/systems/__init__.py:5
Functiondecorator
(cls)
instant-nsr-pl/datasets/__init__.py:5
Functiondecorator
(cls)
instant-nsr-pl/models/__init__.py:5
Functiondisabled_train
Overwrite model.train with this function to make sure train/eval mode does not change anymore.
ldm/modules/encoders/modules.py:178
Functiondisabled_train
Overwrite model.train with this function to make sure train/eval mode does not change anymore.
ldm/models/diffusion/zero123.py:19
Functiondisabled_train
Overwrite model.train with this function to make sure train/eval mode does not change anymore.
ldm/models/diffusion/teeth_dreamer.py:19
Functiondisabled_train
Overwrite model.train with this function to make sure train/eval mode does not change anymore.
ldm/models/diffusion/sync_dreamer.py:19
Functiondownsample_gaussian_blur
(img, ratio)
ldm/base_utils.py:173
Functiondraw_epipolar_lines
(F, img0, img1,num=20)
ldm/base_utils.py:58
Functiondraw_keypoints
(img, kps, colors=None, radius=2)
ldm/base_utils.py:133
Methodencode
(self, *args, **kwargs)
ldm/modules/encoders/modules.py:16
Methodencode
(self, x)
ldm/modules/encoders/modules.py:21
Methodencode
(self, img)
ldm/modules/encoders/modules.py:54
Methodencode
(self, img)
ldm/modules/encoders/modules.py:84
Methodencode
(self, x)
ldm/modules/encoders/modules.py:119
Methodencode
(self, text)
ldm/modules/encoders/modules.py:140
Methodencode
(self, text)
ldm/modules/encoders/modules.py:209
Methodencode
(self, img)
ldm/modules/encoders/modules.py:251
Methodencode
(self, text)
ldm/modules/encoders/modules.py:279
Methodencode
(self, im)
ldm/modules/encoders/modules.py:327
Methodencode
(self, text)
ldm/modules/encoders/modules.py:340
Methodencode
(self, im)
ldm/modules/encoders/modules.py:381
Methodencode
(self, im)
ldm/modules/encoders/modules.py:440
Methodencode
(self, x)
ldm/modules/encoders/modules.py:471
Methodencode
(self, x)
ldm/models/autoencoder.py:269
Methodencode
(self, x, *args, **kwargs)
ldm/models/autoencoder.py:431
Methodencode_to_prequant
(self, x)
ldm/models/autoencoder.py:102
Methodeval
(self)
instant-nsr-pl/models/nerf.py:142
Methodeval
(self)
instant-nsr-pl/models/neus.py:323
Methodexperiment
(self)
instant-nsr-pl/utils/loggers.py:28
Methodexport
(self)
instant-nsr-pl/systems/base.py:115
Methodexport
(self, export_config)
instant-nsr-pl/models/nerf.py:153
Methodexport
(self, export_config)
instant-nsr-pl/models/neus.py:334
Methodforward
(self, batch)
instant-nsr-pl/systems/base.py:25
Methodforward
(self, batch)
instant-nsr-pl/systems/nerf.py:30
Methodforward
(self, inputs, targets, weight=None, reduction='mean')
instant-nsr-pl/systems/criterions.py:11
Methodforward
(self, inputs, targets, valid_mask=None, reduction='mean')
instant-nsr-pl/systems/criterions.py:44
Methodforward
(self, batch)
instant-nsr-pl/systems/neus.py:44
Methodforward
(ctx, x)
instant-nsr-pl/models/utils.py:58
Methodforward
(self, level, threshold=0.)
instant-nsr-pl/models/geometry.py:54
Methodforward
(self, points)
instant-nsr-pl/models/geometry.py:122
Methodforward
(self, points, with_grad=True, with_feature=True, with_laplace=False)
instant-nsr-pl/models/geometry.py:208
Methodforward
(self, rays)
instant-nsr-pl/models/nerf.py:129
Methodforward
(self, features, dirs, *args)
instant-nsr-pl/models/texture.py:23
Methodforward
(self, features, *args)
instant-nsr-pl/models/texture.py:49
Methodforward
(self, x)
instant-nsr-pl/models/neus.py:34
Methodforward
(self, rays)
instant-nsr-pl/models/neus.py:309
Methodforward
(self, x)
instant-nsr-pl/models/network_utils.py:25
Methodforward
(self, x)
instant-nsr-pl/models/network_utils.py:55
Methodforward
(self, x, *args)
instant-nsr-pl/models/network_utils.py:75
Methodforward
(self, x)
instant-nsr-pl/models/network_utils.py:109
Methodforward
(self, x)
instant-nsr-pl/models/network_utils.py:192
Methodforward
(self, x)
ldm/modules/x_transformer.py:34
Methodforward
(self, x, seq_dim=1, offset=0)
ldm/modules/x_transformer.py:45
Methodforward
(self, x, **kwargs)
ldm/modules/x_transformer.py:123
Methodforward
(self, x, **kwargs)
ldm/modules/x_transformer.py:134
Methodforward
(self, x)
ldm/modules/x_transformer.py:146
Methodforward
(self, x)
ldm/modules/x_transformer.py:158
Methodforward
(self, x, residual)
ldm/modules/x_transformer.py:164
Methodforward
(self, x, residual)
ldm/modules/x_transformer.py:173
Methodforward
(self, x)
ldm/modules/x_transformer.py:189
Methodforward
(self, x)
ldm/modules/x_transformer.py:210
Methodforward
( self, x, context=None, mask=None, context_mask=N
ldm/modules/x_transformer.py:268
Methodforward
( self, x, context=None, mask=None, context_mask=N
ldm/modules/x_transformer.py:481
Methodforward
( self, x, return_embeddings=False, mask=None, ret
ldm/modules/x_transformer.py:598
Methodforward
(self, x)
ldm/modules/attention.py:42
Methodforward
(self, x)
ldm/modules/attention.py:51
Methodforward
(self, x)
ldm/modules/attention.py:72
Methodforward
(self, x)
ldm/modules/attention.py:97
Methodforward
(self, x)
ldm/modules/attention.py:135
Methodforward
(self, x, context=None, mask=None)
ldm/modules/attention.py:179
Methodforward
(self, x, context=None)
ldm/modules/attention.py:228
Methodforward
(self, x, context=None)
ldm/modules/attention.py:262
Methodforward
(self, x)
ldm/modules/attention.py:287
Methodforward
(self, x, context=None)
ldm/modules/attention.py:325
Methodforward
(self, x)
ldm/modules/diffusionmodules/openaimodel.py:52
Methodforward
Apply the module to `x` given `emb` timestep embeddings.
ldm/modules/diffusionmodules/openaimodel.py:69
Methodforward
(self, x, emb, context=None)
ldm/modules/diffusionmodules/openaimodel.py:81
Methodforward
(self, x)
ldm/modules/diffusionmodules/openaimodel.py:110
Methodforward
(self,x)
ldm/modules/diffusionmodules/openaimodel.py:131
Methodforward
(self, x)
ldm/modules/diffusionmodules/openaimodel.py:159
← previousnext →501–600 of 848, ranked by callers