Browse Source

print type while printing instance

tags/v0.3.0^2
xuyige 5 years ago
parent
commit
a97bdc7f62
2 changed files with 4 additions and 2 deletions
  1. +3
    -1
      fastNLP/core/instance.py
  2. +1
    -1
      fastNLP/core/losses.py

+ 3
- 1
fastNLP/core/instance.py View File

@@ -30,5 +30,7 @@ class Instance(object):
return self.add_field(name, field)

def __repr__(self):
s = '\''
return "{" + ",\n".join(
"\'" + field_name + "\': " + str(self.fields[field_name]) for field_name in self.fields) + "}"
"\'" + field_name + "\': " + str(self.fields[field_name]) +\
f" type={(str(type(self.fields[field_name]))).split(s)[1]}" for field_name in self.fields) + "}"

+ 1
- 1
fastNLP/core/losses.py View File

@@ -195,7 +195,7 @@ class CrossEntropyLoss(LossBase):

def get_loss(self, pred, target):
return F.cross_entropy(input=pred, target=target,
ignore_index=self.padding_idx)
ignore_index=self.padding_idx)


class L1Loss(LossBase):


Loading…
Cancel
Save