ICLR OSC 2022
Contents:
Introduction
Setup
API Reference
osc.data
osc.loss_global
osc.loss_objects
osc.lr_scheduler
osc.models
osc.train
osc.train.batches_per_epoch_test
osc.train.batches_per_epoch_train
osc.train.batches_per_epoch_val
osc.train.build_dataset_test
osc.train.build_dataset_train
osc.train.build_dataset_val
osc.train.build_dataset_vqa
osc.train.build_linear_probe
osc.train.build_loss_fn_global
osc.train.build_loss_fn_objects
osc.train.build_model
osc.train.build_optimizer
osc.train.build_scheduler
osc.train.extract_vqa_features
osc.train.get_viz_batch
osc.train.log_env_info
osc.train.log_model_parameters
osc.train.main
osc.train.run_test_linear_probes
osc.train.run_test_segmentation
osc.train.run_train_epoch
osc.train.run_train_val_viz_epochs
osc.train.run_val_epoch
osc.train.run_viz
osc.train.update_cfg
osc.train.ModelLoss
osc.utils
osc.viz
osc.wandb_utils
ICLR OSC 2022
»
osc
»
osc.train
»
osc.train.build_linear_probe
View page source
osc.train.build_linear_probe
build_linear_probe
(
cfg
,
output_classes
)
[source]
Return type
Module