snsd0805/Distributed-Training-Example
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

docs: add some hint comment

Browse Source
This commit is contained in:
Ting-Jun Wang 2024-05-15 20:55:08 +08:00
parent e7572347c9
commit 939aa6d92e
Signed by: snsd0805
GPG Key ID: 48D331A3D6160354
2 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
train/dataset.py
View File

@ -1,3 +1,9 @@
'''
Please implement a Dataset which inherit the PyTorch Dataset class.
So that our trainer can load the data from /dataset_dir
'''
import os import os
import numpy as np import numpy as np
from torch.utils.data import Dataset, DataLoader from torch.utils.data import Dataset, DataLoader

5
train/model.py
View File

@ -1,3 +1,8 @@
'''
Please Implement your model here.
'''
from torch import nn from torch import nn
import torch.nn.functional as F import torch.nn.functional as F