projects
/
picoclvr.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
Cosmetics.
[picoclvr.git]
/
main.py
diff --git
a/main.py
b/main.py
index
6d9f69d
..
c01cc8f
100755
(executable)
--- a/
main.py
+++ b/
main.py
@@
-20,7
+20,7
@@
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
######################################################################
parser = argparse.ArgumentParser(
######################################################################
parser = argparse.ArgumentParser(
- description="An implementation of GPT with cache to solve a toy geometric reason
n
ing task."
+ description="An implementation of GPT with cache to solve a toy geometric reasoning task."
)
parser.add_argument("--log_filename", type=str, default="train.log")
)
parser.add_argument("--log_filename", type=str, default="train.log")
@@
-421,9
+421,7
@@
class TaskPicoCLVR(Task):
f"property_{prefix}miss {n_epoch} {100*nb_missing_properties/nb_requested_properties:.02f}%"
)
f"property_{prefix}miss {n_epoch} {100*nb_missing_properties/nb_requested_properties:.02f}%"
)
- img = picoclvr.descr2img(
- result_descr, [0], height=self.height, width=self.width
- )
+ img = picoclvr.descr2img(result_descr, height=self.height, width=self.width)
if img.dim() == 5:
if img.size(1) == 1:
if img.dim() == 5:
if img.size(1) == 1: