Commit d74cda16 authored by Gaëtan Caillaut's avatar Gaëtan Caillaut
Browse files

batch_size=64

parent 35bd6c7f
......@@ -25,9 +25,9 @@ if __name__ == "__main__":
for fold, (train, test) in enumerate(dataset.iter_folds(args.folds, True), 1):
train_loader = DataLoader(
train, collate_fn=collater, shuffle=True, batch_size=128, pin_memory=pm)
train, collate_fn=collater, shuffle=True, batch_size=64, pin_memory=pm)
test_loader = DataLoader(
test, collate_fn=collater, shuffle=False, batch_size=128, pin_memory=pm)
test, collate_fn=collater, shuffle=False, batch_size=64, pin_memory=pm)
if args.jobname is not None:
writer = SummaryWriter(
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment