Add base model and training scripts (!1) · Merge requests · Google Project / Unified Predictor

choucl requested to merge wip/add-base into main Mar 13, 2024

@all I've added the PyTorch version of InceptionTime model in model.py and the basic training script in train.py for the unified model, you could see README.md to find the usage. The scripts could be successfully run on both CPU and GPU (tested with CASGPU1 and CASGPU2). Since I'm not quite familiar with PyTorch. Please feel free to tell me if there's anything that could be improved.

Add base model and training scripts

Merge request reports