exclude some more*2
This commit is contained in:
3
main.py
3
main.py
@ -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)
|
||||
|
Reference in New Issue
Block a user