Add Model class #1126
Add Model class #1126
Conversation
Documentation preview |
edknv
requested review from
marcromeyn and
oliverholworthy
and removed request for
marcromeyn
| ... BinaryOutput(schema.select_by_tag(Tags.TARGET).first), | ||
| ... ) | ||
| ... trainer = Trainer(max_epochs=1) | ||
| ... with Loader(dataset, batch_size=16) as loader: |
There was a problem hiding this comment.
Is it required to use it as a context-manager?
There was a problem hiding this comment.
No, it's probably not necessary in the torch case, but I want to promote this idiom using the context manager everywhere, because the tensorflow equivalent has a memory leak in some cases without a context manager.
| super().__init__() | ||
| self.schema = schema | ||
|
|
||
| self.pre = BlockContainer(name="pre") |
There was a problem hiding this comment.
Should we break this part (until line 77) as a function an use it inside Block as well?
There was a problem hiding this comment.
Yeah, that would be cleaner, but I'm having trouble with torchscript shenanigans.
| """Finds all instances of `ModelOutput` in the model.""" | ||
| return module_utils.find_all_instances(self, ModelOutput) | ||
|
|
||
| def first(self) -> nn.Module: |
There was a problem hiding this comment.
Isn't this part of the Block class as well?
| and len(model_outputs) > 1 | ||
| ): | ||
| raise RuntimeError("Multiple outputs but only one target was provided.") | ||
|
|
There was a problem hiding this comment.
Do we need to check if all model_outputs have a target property set?
There was a problem hiding this comment.
I updated the logic to check if model_outputs has a target when no targets are provided in lines 205-206.
| self, inputs: Union[torch.Tensor, Dict[str, torch.Tensor]], batch: Optional[Batch] = None | ||
| ): | ||
| """Performs a forward pass through the model.""" | ||
| outputs = inputs |
There was a problem hiding this comment.
Same here as for the init, should we break this out in a function so it can be shared between here and the Block class.
merlin/models/torch/models/base.py
Outdated
| def training_step(self, batch, batch_idx): | ||
| """Performs a training step with a single batch.""" | ||
| del batch_idx | ||
| inputs, targets = batch |
There was a problem hiding this comment.
We need some logic here to construct the Batch class and pass it to the forward-pass.
| if self.schema: | ||
| return self.schema | ||
| return Schema([]) | ||
|
|
There was a problem hiding this comment.
We would need to add a method here for output_schema that combines all the output-schema's of the various model-outputs.
There was a problem hiding this comment.
Added an output_schema() method.
2 participants
This PR introduces the
Modelclass for constructing models from individual blocks in the pytorch backend.Some follow-up will be necessary, such as testing with a toy dataset using dataloader,
batch_predict, etc.