exclude some more*2

This commit is contained in:
Si11ium
2019-08-02 19:12:21 +02:00
parent 0a5be5d4da
commit 30525c954e
2 changed files with 3 additions and 2 deletions

View File

@ -123,7 +123,8 @@ if __name__ == '__main__':
print('Epoch {}, total epoches {}'.format(epoch+1, opt.nepoch))
net.train()
# ToDo: We need different dataloader here to train the network in multiple iterations, maybe move the loop down
# for dataloader in ...
for batch_idx, sample in enumerate(dataLoader):
# points: (batch_size, n, 3)
# labels: (batch_size, n)