Browse Source

更新 'train_with_func.py'

master
songyw 1 month ago
parent
commit
230d9fbda7
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      train_with_func.py

+ 1
- 1
train_with_func.py View File

@@ -155,7 +155,7 @@ def train(args):
loader_train = create_loader(
dataset=dataset_train,
batch_size=args.batch_size,
drop_remainder=False,
drop_remainder=args.drop_remainder,
is_training=True,
mixup=args.mixup,
cutmix=args.cutmix,


Loading…
Cancel
Save