mirrors/deit
1
0
Fork 0
mirror of https://github.com/facebookresearch/deit.git synced 2025-06-03 14:52:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

remove distilation loss

Browse Source
This commit is contained in:
ylf 2022-05-31 22:40:07 +08:00
parent c5913bf41f
commit 1b38e41db4
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
engine.py
View File

@ -38,8 +38,8 @@ def train_one_epoch(model: torch.nn.Module, criterion: DistillationLoss,
targets = targets.gt(0.0).type(targets.dtype) targets = targets.gt(0.0).type(targets.dtype)
with torch.cuda.amp.autocast(): with torch.cuda.amp.autocast():
outputs, _, _ = model(samples) loss, _, _ = model(samples)
loss = criterion(samples, outputs, targets) #loss = criterion(samples, outputs, targets)
loss_value = loss.item() loss_value = loss.item()