MCPcopy Create free account

hub / github.com/amazon-science/alexa-arena / functions

Functions501 in github.com/amazon-science/alexa-arena

⨍ Functions 501◇ Types & classes 76
Methodflush
(self)
modeling/vl_model/utils.py:384
Methodforward
(self, x)
modeling/vl_model/layers.py:47
Methodforward
x: b, 512, 26, 26 word: b, 512
modeling/vl_model/layers.py:69
Methodforward
vis: b, 512, h, w txt: b, L, 512 pad_mask: b, L
modeling/vl_model/layers.py:160
Methodforward
vis: 26*26, b, 512 txt: L, b, 512 vis_pos: 26*26, 1, 512 txt_pos: L, 1, 512 pad_mask: b,
modeling/vl_model/layers.py:230
Methodforward
(self, imgs, state)
modeling/vl_model/layers.py:288
Methodforward
img: b, 3, h, w word: b, words word_mask: b, words mask: b, 1, h, w
modeling/vl_model/models.py:81
Methodforward
(self, x)
modeling/vl_model/models.py:196
Methodforward
(self, x)
modeling/vl_model/models.py:263
Methodforward
(self, frames)
modeling/ns_model/nn/enc_visual.py:241
Methodforward
(self, x)
modeling/ns_model/nn/enc_visual.py:247
Methodforward
(self, lang, frames, actions, lens_lang, lens_frames, pos=None)
modeling/ns_model/nn/encodings.py:27
Methodforward
(self, x, tokens)
modeling/ns_model/nn/encodings.py:55
Methodforward
(self, lang, frames, actions, lens_lang, lens_frames)
modeling/ns_model/nn/encodings.py:69
Methodforward
(self, lang, frames, actions)
modeling/ns_model/nn/encodings.py:89
Methodforward
(self, x, pos=None)
modeling/ns_model/nn/encodings.py:115
Methodforward
(self, x, tokens_mask)
modeling/ns_model/nn/encodings.py:131
Methodforward
(self, lang, vocab)
modeling/ns_model/nn/encodings.py:155
Methodforward
pass embedded inputs through embeddings and encode them using a transformer
modeling/ns_model/nn/enc_lang.py:45
Methodforward
(self, inp)
modeling/ns_model/nn/attention.py:20
Methodforward
(self, inp, h)
modeling/ns_model/nn/attention.py:33
Methodforward
pass embedded inputs through embeddings and encode them using a transformer
modeling/ns_model/nn/enc_vl.py:41
Methodforward
(self, x)
modeling/ns_model/nn/dec_object.py:25
Functiongather_jsons
(files, output_path)
modeling/ns_model/utils/data_util.py:85
Functiongenerate_attention_mask
generate mask for attention (a timestep at t does not attend to timesteps after t)
modeling/ns_model/utils/model_util.py:267
Methodget_cdfs
(self)
web_tool/server.py:50
Functionget_coco_kp
(root, image_set, transforms)
modeling/vision_model/coco_utils.py:277
Functionget_feat_shape
Get feat shape depending on the training archi and compress type
modeling/ns_model/utils/data_util.py:177
Functionget_first_available_gpu
()
modeling/vision_model/utils.py:338
Methodget_flask_app
(self)
web_tool/server.py:104
Functionget_last_saved_checkpoint
(exp_save_log_folder)
modeling/vision_model/utils.py:391
Functionget_last_saved_checkpoint
(exp_save_log_folder)
modeling/vl_model/utils.py:160
Methodget_reconstructed_metadata
(self)
arena_wrapper/arena_orchestrator.py:198
Methodget_s3
(self)
modeling/inference/util/utils.py:136
Functionget_task_and_ann_id
single string for task_id and annotation repeat idx
modeling/ns_model/utils/model_util.py:170
Functionget_world_size
()
modeling/vl_model/utils.py:110
Methodglobal_avg
(self)
modeling/vision_model/utils.py:61
Methodglobal_avg
(self)
modeling/vl_model/utils.py:64
Functionhas_interaction
check if low-level action is interactive
modeling/ns_model/utils/model_util.py:159
Functioninference
(test_loader, model, args)
modeling/vl_model/engine.py:170
Functioninit_distributed_mode
(args)
modeling/vision_model/utils.py:307
Functioninit_distributed_mode
(args)
modeling/vl_model/utils.py:131
Functioninit_random_seed
Initialize random seed.
modeling/vl_model/utils.py:170
Methodinit_weights
init linear layers in embeddings
modeling/ns_model/model/base.py:30
FunctionintersectionAndUnionGPU
(output, target, K, threshold=0.5)
modeling/vl_model/utils.py:320
Functionload_log
loading a method json to continue training from the correct place
modeling/ns_model/utils/model_util.py:200
Functionload_model
load pth model from disk
modeling/ns_model/utils/model_util.py:104
Functionload_model_args
load model's args from disk
modeling/ns_model/utils/model_util.py:123
Functionload_model_checkpoint
(model, path_to_model_checkpoint)
modeling/vision_model/train_eval.py:35
Functionmakedirs
(directory)
modeling/inference/util/__init__.py:6
Functionmakedirs
(directory)
arena_wrapper/util/__init__.py:7
Methodmask_to_bbox
(self, mask)
modeling/vision_model/data_generators/arena_data_generator.py:146
Methodmedian
(self)
modeling/vision_model/utils.py:51
Methodmedian
(self)
modeling/vl_model/utils.py:54
Functionmkdir
(path)
modeling/vision_model/utils.py:255
Functionobj_classes_loss
Compute a cross-entropy loss for the object class predictions.
modeling/ns_model/utils/model_util.py:392
Functionparse_metadata_only_bot_position
(metadata, readable_type_matching_dict)
arena_wrapper/util/object_class_decoder.py:400
Methodping
(self)
web_tool/server.py:68
Functionplot_class_counter
(class_counter_path)
modeling/vision_model/prepare_clean_data.py:215
Functionplot_off_the_shelf_metrics
(off_the_shelf_metrics_file, finetuned_metrics_file)
modeling/vision_model/prepare_clean_data.py:279
Functionplot_per_cat_metrics
(per_cat_metrics_file_path, class_counter_path)
modeling/vision_model/prepare_clean_data.py:238
Methodpredict_objects
(self, image, verbose=False)
modeling/ns_model/nn/enc_visual.py:201
Functionprocess
(data_is_valid, object_id_to_class, class_to_area_thresholds, classes_list, split, met_path)
modeling/vision_model/prepare_clean_data.py:19
Functionprocess_traj
(traj_orig, traj_path, r_idx, preprocessor)
modeling/ns_model/utils/data_util.py:69
Methodprocess_utterance
(self)
web_tool/server.py:41
Methodread_cdf
(self, cdf_file_name)
modeling/inference/model_executors/placeholder_model_executor.py:72
Functionread_file_from_s3
Read file from s3 bucket.
modeling/inference/util/utils.py:98
Functionread_image
(path)
modeling/vision_model/data_cleaning_version_control.py:91
Functionread_traj_images
(json_path, image_folder)
modeling/ns_model/utils/data_util.py:35
Functionreduce_dict
Args: input_dict (dict): all the values will be reduced average (bool): whether to do average or sum Reduce the values in the
modeling/vision_model/utils.py:123
Methodrender
()
web_tool/front-end/src/components/DropDown.jsx:72
Methodrender
()
web_tool/front-end/src/components/StreamingContent.jsx:91
Functionrestore_checkpoint
(path_to_model_checkpoint, model, device, optimizer=None)
modeling/inference/util/utils.py:46
Functionsample_batches
sample a batch from each iterator, return Nones if the iterator is empty
modeling/ns_model/utils/data_util.py:160
Functionsave_log
logging a method json for besteffort mode and jobs monitoring on Alex's machine
modeling/ns_model/utils/model_util.py:184
Functionsave_model
save the model to args.dout/model_name or create a symlink from the latest model to args.dout/model_name
modeling/ns_model/utils/model_util.py:131
Functionsave_on_master
(*args, **kwargs)
modeling/vision_model/utils.py:302
Functionsave_on_master
(*args, **kwargs)
modeling/vl_model/utils.py:126
Functionsave_onnx_model
(model, device, onnx_save_path)
modeling/vision_model/utils.py:401
Functionset_random_seed
Set random seed.
modeling/vl_model/utils.py:190
Functionsetup_logger
setup logger for training and testing. Args: save_dir(str): location to save log file distributed_rank(int): device rank when mult
modeling/vl_model/utils.py:394
Methodstart_game
(self)
web_tool/server.py:58
Methodstop
(self)
arena_wrapper/arena_controller.py:179
Methodstop_game
(self)
arena_wrapper/arena_orchestrator.py:81
Methodsynchronize_between_processes
Warning: does not synchronize the deque!
modeling/vision_model/utils.py:37
Methodsynchronize_between_processes
Warning: does not synchronize the deque!
modeling/vl_model/utils.py:40
Functiontensorboard
(writer, metrics, split, iter, frequency, batch_size)
modeling/ns_model/utils/model_util.py:177
Functiontest_extractor
(orig_json_path, extractor, feats_orig)
modeling/ns_model/utils/model_util.py:247
Methodtrain
(self)
modeling/inference/model_executors/placeholder_model_executor.py:32
Methodtrain
(self)
arena_installation_test/arena_nn_model.py:16
Functiontranslate_to_vocab
translate tokens from orig vocab to translate vocab
modeling/ns_model/utils/model_util.py:426
Methodupdate
(self, value, n=1)
modeling/vl_model/utils.py:35
Functionupdate_log
updating a method json for monitoring on Alex's machine
modeling/ns_model/utils/model_util.py:220
Functionupload_folder_to_s3
Upload all files in a folder to bucket.
modeling/inference/util/utils.py:121
Methodvalue
(self)
modeling/vision_model/utils.py:69
Methodvalue
(self)
modeling/vl_model/utils.py:72
Functionwarmup_lr_scheduler
(optimizer, warmup_iters, warmup_factor)
modeling/vision_model/utils.py:244
Functionworker_init_fn
(worker_id, num_workers, rank, seed)
modeling/vl_model/utils.py:217
Functionwrite_file_to_s3
Write file to s3 bucket.
modeling/inference/util/utils.py:107
MethodwsListen
(self)
arena_wrapper/arena_controller.py:106
← previousnext →401–500 of 501, ranked by callers