projects
/
mygpt.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
Update.
[mygpt.git]
/
main.py
diff --git
a/main.py
b/main.py
index
f496e99
..
6c1def7
100755
(executable)
--- a/
main.py
+++ b/
main.py
@@
-155,6
+155,7
@@
class TaskPicoCLVR(Task):
self.train_descr = generate_descr((nb * 4) // 5)
self.test_descr = generate_descr((nb * 1) // 5)
self.train_descr = generate_descr((nb * 4) // 5)
self.test_descr = generate_descr((nb * 1) // 5)
+ # Build the tokenizer
tokens = set()
for d in [ self.train_descr, self.test_descr ]:
for s in d:
tokens = set()
for d in [ self.train_descr, self.test_descr ]:
for s in d:
@@
-215,12
+216,12
@@
class TaskPicoCLVR(Task):
img = [ picoclvr.descr2img(d, height = self.height, width = self.width) for d in descr ]
img = torch.cat(img, 0)
img = [ picoclvr.descr2img(d, height = self.height, width = self.width) for d in descr ]
img = torch.cat(img, 0)
-
fil
e_name = f'result_picoclvr_{n_epoch:04d}.png'
+
imag
e_name = f'result_picoclvr_{n_epoch:04d}.png'
torchvision.utils.save_image(
img / 255.,
torchvision.utils.save_image(
img / 255.,
-
fil
e_name, nrow = nb_per_primer, pad_value = 0.8
+
imag
e_name, nrow = nb_per_primer, pad_value = 0.8
)
)
- log_string(f'wrote {
fil
e_name}')
+ log_string(f'wrote {
imag
e_name}')
nb_missing = sum( [
x[2] for x in picoclvr.nb_missing_properties(
nb_missing = sum( [
x[2] for x in picoclvr.nb_missing_properties(