multiml.agent.pytorch.pytorch_sposnas module

class multiml.agent.pytorch.pytorch_sposnas.PytorchSPOSNASAgent(training_choiceblock_model=True, **kwargs)

Bases: PytorchConnectionRandomSearchAgent

Agent packing subtasks using Pytorch SPOS-NAS Model.

__init__(training_choiceblock_model=True, **kwargs)
Parameters:

  • training_choiceblock_model (bool) – Training choiceblock model after connecting submodels

  • **kwargs – Arbitrary keyword arguments

execute()

Execute.