Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Sign in / Register
Toggle navigation
Menu
Open sidebar
Nianchen Deng
deeplightfield
Commits
f6604bd2
Commit
f6604bd2
authored
Mar 16, 2021
by
Nianchen Deng
Browse files
rebuttal version
parent
6e54b394
Changes
21
Hide whitespace changes
Inline
Side-by-side
run_spherical_view_syn.py
View file @
f6604bd2
...
...
@@ -6,9 +6,6 @@ import torch.optim
from
tensorboardX
import
SummaryWriter
from
torch
import
nn
sys
.
path
.
append
(
os
.
path
.
abspath
(
sys
.
path
[
0
]
+
'/../'
))
__package__
=
"deep_view_syn"
parser
=
argparse
.
ArgumentParser
()
parser
.
add_argument
(
'--device'
,
type
=
int
,
default
=
3
,
help
=
'Which CUDA device to use.'
)
...
...
@@ -46,15 +43,15 @@ if opt.res:
torch
.
cuda
.
set_device
(
opt
.
device
)
print
(
"Set CUDA:%d as current device."
%
torch
.
cuda
.
current_device
())
from
.
my
import
netio
from
.
my
import
util
from
.
my
import
device
from
.
my
import
loss
from
.
my.progress_bar
import
progress_bar
from
.
my.simple_perf
import
SimplePerf
from
.
data.spherical_view_syn
import
*
from
.
data.loader
import
FastDataLoader
from
.
configs.spherical_view_syn
import
SphericalViewSynConfig
from
my
import
netio
from
my
import
util
from
my
import
device
from
my
import
loss
from
my.progress_bar
import
progress_bar
from
my.simple_perf
import
SimplePerf
from
data.spherical_view_syn
import
*
from
data.loader
import
FastDataLoader
from
configs.spherical_view_syn
import
SphericalViewSynConfig
config
=
SphericalViewSynConfig
()
...
...
Prev
1
2
Next
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment