Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/atomicarchitects/equiformer_v2
/ functions
Functions
244 in github.com/atomicarchitects/equiformer_v2
⨍
Functions
244
◇
Types & classes
55
Method
forward
(self, input_embedding, output_scale=None)
nets/equiformer_v2/so3.py:595
Method
forward
(self, input_embedding)
nets/equiformer_v2/so3.py:649
Method
forward
( self, atomic_numbers, edge_distance, edge_index )
nets/equiformer_v2/input_block.py:78
Method
forward
( self, x, atomic_numbers, edge_distance, edge_index )
nets/equiformer_v2/transformer_block.py:231
Method
forward
(self, input_embedding)
nets/equiformer_v2/transformer_block.py:415
Method
forward
( self, x, # SO3_Embedding atomic_numbers, edge_distance,
nets/equiformer_v2/transformer_block.py:583
Method
forward
Assume input is of shape [N, sphere_basis, C]
nets/equiformer_v2/layer_norm.py:61
Method
forward
Assume input is of shape [N, sphere_basis, C]
nets/equiformer_v2/layer_norm.py:153
Method
forward
Assume input is of shape [N, sphere_basis, C]
nets/equiformer_v2/layer_norm.py:228
Method
forward
Assume input is of shape [N, sphere_basis, C]
nets/equiformer_v2/layer_norm.py:311
Method
forward
(self, node_input)
nets/equiformer_v2/layer_norm.py:374
Method
forward
(self, inputs)
nets/equiformer_v2/radial_function.py:29
Method
forward
(self, inputs)
nets/equiformer_v2/activation.py:15
Method
forward
(self, inputs)
nets/equiformer_v2/activation.py:36
Method
forward
(self, inputs)
nets/equiformer_v2/activation.py:55
Method
forward
(self, x)
nets/equiformer_v2/activation.py:70
Method
forward
(self, x)
nets/equiformer_v2/activation.py:87
Method
forward
(self, x)
nets/equiformer_v2/activation.py:101
Method
forward
`gating_scalars`: shape [N, lmax * num_channels] `input_tensors`: shape [N, (lmax + 1) ** 2, num_channels]
nets/equiformer_v2/activation.py:130
Method
forward
(self, inputs, SO3_grid)
nets/equiformer_v2/activation.py:162
Method
forward
(self, input_scalars, input_tensors, SO3_grid)
nets/equiformer_v2/activation.py:182
Method
from_grid
(self, grid, lmax, mmax)
nets/equiformer_v2/so3.py:575
Function
get_data
(index)
oc20/trainer/make_lmdb_sizes.py:24
Function
get_world_size
()
utils.py:25
Function
init_distributed_mode
(args)
utils.py:46
Method
load_checkpoint
(self, checkpoint_path)
oc20/trainer/base_trainer_oc20.py:393
Method
load_task
(self)
oc20/trainer/forces_trainer_v2.py:116
Method
load_task
(self)
oc20/trainer/energy_trainer_v2.py:128
Method
multiply
(obj, num)
oc20/trainer/base_trainer_v2.py:443
Function
multistep_lr_lambda
(current_step, scheduler_params)
oc20/trainer/lr_scheduler.py:57
Method
no_op
(*args, **kwargs)
logger.py:56
Method
no_op
(*args, **kwargs)
oc20/trainer/logger.py:56
Method
num_params
(self)
nets/equiformer_v2/equiformer_v2_oc20.py:479
Method
run
(self)
oc20/trainer/task_compute_stats.py:7
Function
save_on_master
(*args, **kwargs)
utils.py:41
Function
setup
(config)
oc20/trainer/dist_setup.py:14
Method
to_grid
(self, SO3_grid, lmax=-1)
nets/equiformer_v2/so3.py:367
Method
train
(self, disable_eval_tqdm=False)
oc20/trainer/forces_trainer_v2.py:295
Method
train
(self, disable_eval_tqdm=False)
oc20/trainer/energy_trainer_v2.py:227
Function
train_one_epoch
(model: torch.nn.Module, criterion: torch.nn.Module, norm_factor: list,
engine.py:30
Function
train_one_epoch
(model: torch.nn.Module, criterion: torch.nn.Module, norm_factor: list,
oc20/trainer/engine.py:30
Method
validate
(self, split="val", disable_tqdm=False)
oc20/trainer/base_trainer_oc20.py:612
Method
verbose
(self, *args)
logger.py:46
Method
verbose
(self, *args)
oc20/trainer/logger.py:46
← previous
201–244 of 244, ranked by callers