Releases
v23.04.00
What’s Changed
🐜 Bug Fixes
Update multi-gpu notebook to set cupy device @edknv (#675 )
Fix bug in get_output_sizes_from_schema with core-schema @marcromeyn (#663 )
Remove torch.squeeze() step from the model's forward method. @sararb (#659 )
Set device in dataloaders @edknv (#654 )
Fix the predictions returned by Trainer.predict(..)
@sararb (#641 )
🚀 Features
📄 Documentation
extend getting started serving example to serve NVT and TF4Rec model together @rnyak (#670 )
Cropped the table and addressed comments from previous PR @nzarif (#510 )
Update example notebooks to create schema object from merlin core @rnyak (#650 )
replace NVTabulardataloader with Merlindataloader @rnyak (#644 )
🔧 Maintenance
Update multi-gpu notebook to set cupy device @edknv (#675 )
add concurrency setting to stop tests when new commits get pushed to a PR @nv-alaiacano (#673 )
Switch to using 2 GPU action runners for multi-GPU testing @karlhigley (#665 )
Add workflow to check if base branch of pull request is development @oliverholworthy (#656 )
fix the ci script for new unit tests setup @jperez999 (#658 )
Add unit test for serving torchscript model example notebook @rnyak (#657 )
Separate notebook tests into their own tox environment @nv-alaiacano (#653 )
Update usage of use_amp
to use_cuda_amp
for transformers>=4.20 @oliverholworthy (#627 )
Update example notebooks to create schema object from merlin core @rnyak (#650 )
Update padding of ragged features to enable dataloader change @oliverholworthy (#647 )
fix model output_schema dims for BC/Regression task case @rnyak (#646 )
fix test_remove_consecutive_interactions unit test @rnyak (#643 )
replace NVTabulardataloader with Merlindataloader @rnyak (#644 )
Cleanup shapes in model.input_schema and output_schema @rnyak (#628 )
Migrate schema Tags to merlin.schema.Tags @nv-alaiacano (#632 )
Clean up imports in tests @marcromeyn (#626 )
You can’t perform that action at this time.