MCPcopy Create free account

hub / github.com/Yasoz/DiffTraj / functions

Functions61 in github.com/Yasoz/DiffTraj

⨍ Functions 61◇ Types & classes 12
↓ 16 callersMethodformat
(self, record)
utils/logger.py:61
↓ 7 callersMethod__init__
(self, config)
utils/Traj_UNet.py:239
↓ 5 callersFunctionnonlinearity
(x)
utils/Traj_UNet.py:75
↓ 4 callersFunctionNormalize
(in_channels)
utils/Traj_UNet.py:80
↓ 3 callersMethodstate_dict
(self)
utils/EMA.py:48
↓ 2 callersFunctioncompute_alpha
(beta, t)
utils/utils.py:60
↓ 2 callersFunctiongather
Gather consts for $t$ and reshape to feature map shape
main.py:27
↓ 2 callersFunctiongather
Gather consts for $t$ and reshape to feature map shape :param consts: (N, 1, 1) :param t: (N, H, W) :return: (N, H, W)
utils/utils.py:39
↓ 2 callersMethodload_state_dict
(self, state_dict)
utils/EMA.py:51
↓ 1 callersFunctionNormalize
(in_channels)
utils/module.py:32
↓ 1 callersMethod__init__
(self, name="default", colorize=False, log_path=None,
utils/logger.py:69
↓ 1 callersMethod_process
(msg, loglevel, colorize)
utils/logger.py:36
↓ 1 callersMethodadd_handler
(self, stream)
utils/logger.py:130
↓ 1 callersMethodclear_handlers
(self)
utils/logger.py:144
↓ 1 callersMethodema
(self, module)
utils/EMA.py:27
↓ 1 callersFunctionget_timestep_embedding
(timesteps, embedding_dim)
utils/Traj_UNet.py:9
↓ 1 callersFunctionlog_info
output the information about model
utils/logger.py:166
↓ 1 callersFunctionmain
(config, logger, exp_dir)
main.py:33
↓ 1 callersFunctionq_xt_x0
(x0, t)
main.py:36
↓ 1 callersMethodregister
(self, module)
utils/EMA.py:11
↓ 1 callersMethodsetLevel
(self, level: Union[str, int])
utils/logger.py:122
↓ 1 callersMethodupdate
(self, module)
utils/EMA.py:18
Method__init__
(self, embedding_dim)
utils/Traj_UNet.py:24
Method__init__
(self, embedding_dim=128, hidden_dim=256)
utils/Traj_UNet.py:37
Method__init__
(self, in_channels, with_conv=True)
utils/Traj_UNet.py:88
Method__init__
(self, in_channels, with_conv=True)
utils/Traj_UNet.py:108
Method__init__
(self, in_channels, out_channels=None, conv_shortcut=False,
utils/Traj_UNet.py:130
Method__init__
(self, in_channels)
utils/Traj_UNet.py:191
Method__init__
(self, config)
utils/Traj_UNet.py:403
Method__init__
(self, colorize=False, *args, **kwargs)
utils/logger.py:31
Method__init__
(self, in_channels)
utils/module.py:40
Method__init__
(self, mu=0.999)
utils/EMA.py:7
Functiondistance
Calculate the great circle distance between two points on the earth (specified in decimal degrees)
utils/utils.py:90
Functiondivide_grids
(boundary, grids_num)
utils/utils.py:77
Methodema_copy
(self, module)
utils/EMA.py:34
Functionextract
(input, t, x)
utils/helper.py:8
Methodforward
(self, x)
utils/Traj_UNet.py:28
Methodforward
(self, attr)
utils/Traj_UNet.py:50
Methodforward
(self, x)
utils/Traj_UNet.py:98
Methodforward
(self, x)
utils/Traj_UNet.py:119
Methodforward
(self, x, temb)
utils/Traj_UNet.py:170
Methodforward
(self, x)
utils/Traj_UNet.py:217
Methodforward
(self, x, t, extra_embed=None)
utils/Traj_UNet.py:348
Methodforward
(self, x, t, attr)
utils/Traj_UNet.py:415
Methodforward
(self, x)
utils/module.py:66
Methodfunc_wrapper
(*args, **kwargs)
utils/logger.py:110
Methodget_handlers
(self)
utils/logger.py:148
Functionget_timestep_embedding
This matches the implementation in Denoising Diffusion Probabilistic Models: From Fairseq. Build sinusoidal embeddings. This matches
utils/module.py:6
Methodinner_formatter
(self)
utils/logger.py:162
Methodinner_logger
(self)
utils/logger.py:154
Methodinner_stream_handler
(self)
utils/logger.py:158
Methodlog_function
(self)
utils/logger.py:108
Functionmake_beta_schedule
(schedule='linear', n_timesteps=1000, start=1e-5, end=1e-2)
utils/helper.py:14
Functionnonlinearity
(x)
utils/module.py:27
Functionp_xt
(xt, noise, t, next_t, beta, eta=0)
utils/utils.py:66
Functionq_xt_x0
(x0, t, alpha_bar)
utils/utils.py:50
Methodremove_handler
(self, stream)
utils/logger.py:137
Functionresample_trajectory
Resamples a trajectory to a new length. Parameters: x (np.ndarray): original trajectory, shape (N, 2) length (int): length o
utils/utils.py:6
Functionsample_batch
(size, noise=1.0)
utils/helper.py:4
Functiontime_warping
Resamples a trajectory to a new length.
utils/utils.py:26
Methodwrapper
(func)
utils/logger.py:109