mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix error & add nonfinite warning

Browse Source
This commit is contained in:
James Betker 2021-11-09 23:58:41 -07:00
parent 5d5558893a
commit 79367f753d
4 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
codes/models/classifiers/resnet_with_checkpointing.py
View File

@ -3,7 +3,6 @@
import torch import torch
import torch.nn as nn import torch.nn as nn
from torchvision.models.resnet import BasicBlock, Bottleneck from torchvision.models.resnet import BasicBlock, Bottleneck
from torchvision.models.utils import load_state_dict_from_url
import torchvision import torchvision

1
codes/models/classifiers/weighted_conv_resnet.py
View File

@ -2,7 +2,6 @@ import torch
import torchvision import torchvision
from torch import Tensor from torch import Tensor
import torch.nn as nn import torch.nn as nn
from torchvision.models.utils import load_state_dict_from_url
from typing import Type, Any, Callable, Union, List, Optional, OrderedDict, Iterator from typing import Type, Any, Callable, Union, List, Optional, OrderedDict, Iterator
__all__ = ['ResNet', 'resnet18', 'resnet34', 'resnet50', 'resnet101', __all__ = ['ResNet', 'resnet18', 'resnet34', 'resnet50', 'resnet101',

1
codes/models/segformer/backbone.py
View File

@ -3,7 +3,6 @@
import torch import torch
import torch.nn as nn import torch.nn as nn
from torchvision.models.resnet import BasicBlock, Bottleneck from torchvision.models.resnet import BasicBlock, Bottleneck
from torchvision.models.utils import load_state_dict_from_url
import torchvision import torchvision

2
codes/trainer/steps.py
View File

@ -227,6 +227,8 @@ class ConfigurableStep(Module):
new_state.update(lstate) new_state.update(lstate)
else: else:
l = loss(self.get_network_for_name(self.step_opt['training']), local_state) l = loss(self.get_network_for_name(self.step_opt['training']), local_state)
if not l.isfinite():
print(f'!!Detected non-finite loss {loss_name}')
total_loss += l * self.weights[loss_name] total_loss += l * self.weights[loss_name]
# Record metrics. # Record metrics.
if isinstance(l, torch.Tensor): if isinstance(l, torch.Tensor):