Laurent/REVA-QCAV
1
0
Fork 0
mirror of https://github.com/Laurent2916/REVA-QCAV.git synced 2024-11-09 23:12:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

Set drop_last=True for val_loader

Browse source 
Former-commit-id: adb1d7e9348a2707f38d9b86f57bb3ae2cbc2b73
This commit is contained in:
whenyd 2020-03-13 11:21:16 +08:00 committed by yangdong
parent d292e8c6cd
commit cf505c6549
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
train.py
View file

@ -35,7 +35,7 @@ def train_net(net,
n_train = len(dataset) - n_val n_train = len(dataset) - n_val
train, val = random_split(dataset, [n_train, n_val]) train, val = random_split(dataset, [n_train, n_val])
train_loader = DataLoader(train, batch_size=batch_size, shuffle=True, num_workers=8, pin_memory=True) train_loader = DataLoader(train, batch_size=batch_size, shuffle=True, num_workers=8, pin_memory=True)
val_loader = DataLoader(val, batch_size=batch_size, shuffle=False, num_workers=8, pin_memory=True) val_loader = DataLoader(val, batch_size=batch_size, shuffle=False, num_workers=8, pin_memory=True, drop_last=True)
writer = SummaryWriter(comment=f'LR_{lr}_BS_{batch_size}_SCALE_{img_scale}') writer = SummaryWriter(comment=f'LR_{lr}_BS_{batch_size}_SCALE_{img_scale}')
global_step = 0 global_step = 0