Object Detection

class olympus.tasks.detection.ObjectDetection(detector, optimizer, lr_scheduler, dataloader, criterion=None, device=None, storage=None)

Bases: olympus.tasks.task.Task

Attributes:

device

events

metrics

model

Methods

eval_loss(batch)	This is used to compute validation and test loss
fit(epochs[, context])	Execute a single batch
get_space()	Return hyper parameter space
init([optimizer, lr_schedule, model, uid])	Used to initialize the hyperparameters is any
load_state_dict(state[, strict])	Try to load a previous unfinished state to resume
state_dict([destination, prefix, keep_vars])	Save a state the task can go back to if an error occur

epoch
parameters
report
resumed
set_device
step
summary

epoch(epoch, context)

eval_loss(batch)

This is used to compute validation and test loss

fit(epochs, context=None)

Execute a single batch

Parameters:

epoch: int

current step in the training process

context: dict

Optional Context

Notes

You should wrap whatever code you have here inside a BadResumeGuard to prevent users from resuming a failed task that can have a bad states

To resume a task, you need to create a clean one with the same hyper parameters. It will pickup automatically where at its last checkpoint

get_space()

Return hyper parameter space

init(optimizer=None, lr_schedule=None, model=None, uid=None)

Used to initialize the hyperparameters is any

load_state_dict(state, strict=True)

Try to load a previous unfinished state to resume

Notes

You should wrap whatever code you have here inside a BadResumeGuard to prevent users from resuming a failed task that can have a bad states

To resume a task, you need to create a clean one with the same hyper parameters. It will pickup automatically where at its last checkpoint

model

parameters()

state_dict(destination=None, prefix='', keep_vars=False)

Save a state the task can go back to if an error occur

step(step, input, context)