Commit 8697f9e0 authored by Surgan Jandial's avatar Surgan Jandial Committed by Soumith Chintala
Browse files

Making references/classification/train.py and...

Making references/classification/train.py and references/classification/utils.py  compatible with python2  (#831)

* linter fixes

* linter fixes
parent 742fd13c
from __future__ import print_function
import datetime
import os
import time
......
from __future__ import print_function
from collections import defaultdict, deque
import datetime
import time
......
......@@ -33,7 +33,6 @@ class InvertedResidual(nn.Module):
])
self.conv = nn.Sequential(*layers)
def forward(self, x):
if self.use_res_connect:
return x + self.conv(x)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment