mirrors/PaddleClas
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleClas.git synced 2025-06-03 21:55:06 +08:00
Code Issues Projects Releases Wiki Activity

refine docstring

Browse Source
This commit is contained in:
HydrogenSulfate 2022-08-18 20:20:07 +08:00
parent 84870b771d
commit a646adf754
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ppcls/data/dataloader/pk_sampler.py
View File

@ -29,8 +29,8 @@ class PKSampler(DistributedBatchSampler):
Args:
dataset (Dataset): Dataset which contains list of (img_path, pid, camid))
batch_size (_type_): batch size
sample_per_id (_type_): number of instance(s) within an class
batch_size (int): batch size
sample_per_id (int): number of instance(s) within an class
shuffle (bool, optional): _description_. Defaults to True.
drop_last (bool, optional): whether to discard the data at the end. Defaults to True.
sample_method (str, optional): sample method when generating prob_list. Defaults to "sample_avg_prob".