HTensor/IGEV
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

todo

Browse Source
This commit is contained in:
HTensor 2023-04-19 00:58:31 +08:00
parent 54a2eabbff
commit 93371a3c4d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
IGEV-Stereo/train_stereo.py
View File

@ -132,7 +132,7 @@ class Logger:
def train(args): def train(args):
# todo: compile the model to speed up at pytorch 2.0.
model = nn.DataParallel(IGEVStereo(args)) model = nn.DataParallel(IGEVStereo(args))
print("Parameter Count: %d" % count_parameters(model)) print("Parameter Count: %d" % count_parameters(model))