projects
/
picoclvr.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
Update.
[picoclvr.git]
/
tasks.py
diff --git
a/tasks.py
b/tasks.py
index
0ab1823
..
2c2f914
100755
(executable)
--- a/
tasks.py
+++ b/
tasks.py
@@
-1459,8
+1459,7
@@
class Grid(Task):
nb_train_samples,
nb_test_samples,
batch_size,
nb_train_samples,
nb_test_samples,
batch_size,
- height,
- width,
+ size,
logger=None,
device=torch.device("cpu"),
):
logger=None,
device=torch.device("cpu"),
):
@@
-1468,7
+1467,7
@@
class Grid(Task):
self.device = device
self.batch_size = batch_size
self.device = device
self.batch_size = batch_size
- self.grid_factory = grid.GridFactory(
height=height, width=width
)
+ self.grid_factory = grid.GridFactory(
size=size
)
if logger is not None:
logger(
if logger is not None:
logger(