Skip to content
Projects
Groups
Snippets
Help
This project
Loading...
Sign in / Register
Toggle navigation
S
SCL-my
Overview
Overview
Details
Activity
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
YuxuanGuo
SCL-my
Commits
84297604
Commit
84297604
authored
Mar 10, 2022
by
yuxguo
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
fix
parent
80ee6423
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
3 deletions
+3
-3
main.py
+3
-3
No files found.
main.py
View file @
84297604
...
@@ -355,9 +355,9 @@ def get_dataloader():
...
@@ -355,9 +355,9 @@ def get_dataloader():
valid
=
RAVENdataset
(
args
.
dataset_path
,
"val"
,
args
.
val_figure_configurations
,
ORIGIN_IMAGE_SIZE
,
transform
=
transforms
.
Compose
([
ToTensor
()]))
valid
=
RAVENdataset
(
args
.
dataset_path
,
"val"
,
args
.
val_figure_configurations
,
ORIGIN_IMAGE_SIZE
,
transform
=
transforms
.
Compose
([
ToTensor
()]))
test
=
RAVENdataset
(
args
.
dataset_path
,
"test"
,
args
.
test_figure_configurations
,
ORIGIN_IMAGE_SIZE
,
transform
=
transforms
.
Compose
([
ToTensor
()]))
test
=
RAVENdataset
(
args
.
dataset_path
,
"test"
,
args
.
test_figure_configurations
,
ORIGIN_IMAGE_SIZE
,
transform
=
transforms
.
Compose
([
ToTensor
()]))
trainloader
=
DataLoader
(
train
,
batch_size
=
args
.
batch_size
,
shuffle
=
True
,
num_workers
=
args
.
load
_workers
)
trainloader
=
DataLoader
(
train
,
batch_size
=
args
.
batch_size
,
shuffle
=
True
,
num_workers
=
args
.
num
_workers
)
validloader
=
DataLoader
(
valid
,
batch_size
=
args
.
batch_size
,
shuffle
=
False
,
num_workers
=
args
.
load
_workers
)
validloader
=
DataLoader
(
valid
,
batch_size
=
args
.
batch_size
,
shuffle
=
False
,
num_workers
=
args
.
num
_workers
)
testloader
=
DataLoader
(
test
,
batch_size
=
args
.
batch_size
,
shuffle
=
False
,
num_workers
=
args
.
load
_workers
)
testloader
=
DataLoader
(
test
,
batch_size
=
args
.
batch_size
,
shuffle
=
False
,
num_workers
=
args
.
num
_workers
)
return
trainloader
,
validloader
,
testloader
return
trainloader
,
validloader
,
testloader
...
...
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