There is a newer version of the record available.

Published March 12, 2021 | Version v0.3.0
Software Open

phborba/pytorch_segmentation_models_trainer: Version 0.3.0

Authors/Creators

Description

  • Custom metric option in the model config;
  • pytorch_toolbelt added as required package. This enables usage of the models, losses and metrics in the training;
  • Added the option of setting a limit of rows to be read in the csv dataset;
  • Added the option of setting a root_dir to the dataset. This root_dir will be concatenated to the entry in the csv dataset before loading the image;
  • Bug fixes on image_callback;

Files

phborba/pytorch_segmentation_models_trainer-v0.3.0.zip

Files (3.8 MB)

Additional details

Related works

Is supplement to
https://github.com/phborba/pytorch_segmentation_models_trainer/tree/v0.3.0 (URL)