issues
search
boostcampaitech6
/
level1-imageclassification-cv-10
level1-imageclassification-cv-10 created by GitHub Classroom
1
stars
2
forks
source link
feat: add sampler base code
#12
Closed
lukehanjun
closed
8 months ago
lukehanjun
commented
8 months ago
Explanation
train.py 파일에 sampler 관련 코드를 추가하였습니다.
torchsampler의 ImbalancedDatasetSampler, 그리고 torch.utils의 WeightedRandomSampler를 추가하였습니다.
args 파일에 우희님께서 이미 추가해주신 sampler를 지정하고 train.py를 실행시키시면 됩니다.
Explanation