UniTrain
UniTrain copied to clipboard
wncc
UniTrain is an open-source, unified platform for effortless machine learning model training, evaluation, and deployment across diverse tasks. Experience seamless experimentation and model deployment w...
Make a new file named 'trainClassification.py' inside the 'training' folder and add this feature along with the training part from the readme.md
Adds the V-Net model for classification. Followed [mattmacy/vnet.pytorch](https://github.com/mattmacy/vnet.pytorch). Solves #11
NOTE: Making a new branch for this issues is compulsory before merging to stg-dev. You have to make three new files: `models/new_model.py utils/new_utils.py dataset/new_dataset.py`
