Skip to main content
Back to top
Ctrl
+
K
SODa v0.4.0 documentation
Search
Ctrl
+
K
Содержание
Структура проекта
Параметры запуска
Генерация моделей
API
engine
engine.data
engine.data.DataModule
engine.model
engine.model.Model
engine.trainer
engine.trainer.Trainer
models
models.generator
models.generator.BackboneGen
models.generator.BaseConfig
models.generator.BlockGen
models.generator.Head
models.generator.HeadGen
models.generator.ModelGen
models.generator.NeckGen
models.modules
models.modules.Conv
models.modules.ConvLSTM
models.modules.Dense
models.modules.LI
models.modules.LIF
models.modules.LSTM
models.modules.LayerGen
models.modules.Norm
models.modules.Pass
models.modules.Pool
models.modules.ReLU
models.modules.Residual
models.modules.Return
models.modules.SiLU
models.modules.Storage
models.modules.SumPool2d
models.modules.Tanh
models.modules.Up
models.soda
models.soda.SODa
models.yolo
models.yolo.Yolo
run
run.eval
run.eval.eval_spin
run.interactive
run.interactive.interactive_spin
run.train
run.train.train_spin
utils
utils.anchors
utils.anchors.AnchorGenerator
utils.box
utils.box.box_center_to_corner
utils.box.box_corner_to_center
utils.box.box_iou
utils.box.multibox_detection
utils.box.nms
utils.box.offset_boxes
utils.box.offset_inverse
utils.datasets
utils.datasets.MTProphesee
utils.datasets.MTPropheseeDataset
utils.datasets.PropheseeDataModule
utils.datasets.PropheseeDatasetBase
utils.datasets.STProphesee
utils.datasets.STPropheseeDataset
utils.devices
utils.devices.cpu
utils.devices.gpu
utils.devices.num_gpus
utils.devices.try_all_gpus
utils.devices.try_gpu
utils.evaluate
utils.evaluate.SODAeval
utils.model_loader
utils.model_loader.ModelLoader
utils.plotter
utils.plotter.Plotter
utils.progress_board
utils.progress_board.ProgressBoard
utils.roi
utils.roi.RoI
.rst
.pdf
utils.devices.gpu
Contents
gpu()
utils.devices.gpu
#
gpu
(
i
:
int
=
0
)
→
device
[source]
#
Get a GPU device
Contents
gpu()