Safran/KPConv-PyTorch
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

.

Browse source
This commit is contained in:
HuguesTHOMAS 2021-09-27 11:30:10 -04:00
parent cf1f06381e
commit 73e444d486
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
utils/trainer.py
View file

@ -198,6 +198,9 @@ class ModelTrainer:
#torch.nn.utils.clip_grad_norm_(net.parameters(), config.grad_clip_norm) #torch.nn.utils.clip_grad_norm_(net.parameters(), config.grad_clip_norm)
torch.nn.utils.clip_grad_value_(net.parameters(), config.grad_clip_norm) torch.nn.utils.clip_grad_value_(net.parameters(), config.grad_clip_norm)
self.optimizer.step() self.optimizer.step()
torch.cuda.empty_cache()
torch.cuda.synchronize(self.device) torch.cuda.synchronize(self.device)
t += [time.time()] t += [time.time()]