Templates finalized

2020-05-17 22:05:20 +02:00
parent 398504aee6
commit fc93f71608
20 changed files with 459 additions and 166 deletions
--- a/_templates/new_project/datasets/template_dataset.py
+++ b/_templates/new_project/datasets/template_dataset.py
@ -0,0 +1,6 @@
+from torch.utils.data import Dataset
+
+
+class TemplateDataset(Dataset):
+    def __init__(self, *args, **kwargs):
+        super(TemplateDataset, self).__init__()