↓ 9 callersMethod__init__(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
Pretrain/models/Point_MAE.py:86
↓ 9 callersFunctionevaluate(
args,
curr_epoch,
model,
criterion,
dataset_config,
dataset_loader,
logger,
Downstream/3detr/engine.py:143
↓ 9 callersFunctionsave_checkpoint(
checkpoint_dir,
model_no_ddp,
optimizer,
epoch,
args,
best_val_metrics,
filename
Downstream/3detr/utils/io.py:8
↓ 5 callersMethod__init__(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
Pretrain/models/multimae_utils.py:139
↓ 4 callersMethod__init__(self, d_model=256, nhead=8, num_encoder_layers=6,
num_decoder_layers=6, dim_feedforward=2048
Downstream/detr/models/transformer_orig.py:21
↓ 4 callersMethod__init__(self, d_model=256, nhead=8, num_encoder_layers=6,
num_decoder_layers=6, dim_feedforward=2048
Downstream/detr/models/transformer.py:21