Skip to content

Commit 4a0f341

Browse files
authored
[mnist-acgan] Upgrade to tfjs and tfjs-node 1.3.2 (#340)
- In the latest version, `trainOnBatch()` is async, unlike the sync method in the old version.
1 parent c58eb9e commit 4a0f341

File tree

3 files changed

+64
-75
lines changed

3 files changed

+64
-75
lines changed

mnist-acgan/gan.js

+2-2
Original file line numberDiff line numberDiff line change
@@ -356,8 +356,8 @@ async function trainCombinedModelOneStep(batchSize, latentSize, combined) {
356356
return [zVectors, sampledLabels, trick];
357357
});
358358

359-
const losses =
360-
combined.trainOnBatch([noise, sampledLabels], [trick, sampledLabels]);
359+
const losses = await combined.trainOnBatch(
360+
[noise, sampledLabels], [trick, sampledLabels]);
361361
tf.dispose([noise, sampledLabels, trick]);
362362
return losses;
363363
}

mnist-acgan/package.json

+3-3
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99
"node": ">=8.11.0"
1010
},
1111
"dependencies": {
12-
"@tensorflow/tfjs": "1.2.2"
12+
"@tensorflow/tfjs": "1.3.2"
1313
},
1414
"scripts": {
1515
"build": "cross-env NODE_ENV=production parcel build index.html --no-minify --public-url ./",
@@ -20,8 +20,8 @@
2020
"devDependencies": {
2121
"@babel/core": "^7.0.0-0",
2222
"@babel/plugin-transform-runtime": "^7.1.0",
23-
"@tensorflow/tfjs-node": "^1.2.3",
24-
"@tensorflow/tfjs-node-gpu": "^1.2.3",
23+
"@tensorflow/tfjs-node": "^1.3.2",
24+
"@tensorflow/tfjs-node-gpu": "^1.3.2",
2525
"argparse": "^1.0.10",
2626
"babel-core": "^6.26.3",
2727
"babel-plugin-transform-runtime": "~6.23.0",

mnist-acgan/yarn.lock

+59-70
Original file line numberDiff line numberDiff line change
@@ -652,73 +652,73 @@
652652
resolved "https://registry.yarnpkg.com/@nodelib/fs.stat/-/fs.stat-1.1.3.tgz#2b5a3ab3f918cca48a8c754c08168e3f03eba61b"
653653
integrity sha512-shAmDyaQC4H92APFoIaVDHCx5bStIocgvbwQyxPRrbUY20V1EYTbSDchWbuwlMG3V17cprZhA6+78JfB+3DTPw==
654654

655-
"@tensorflow/tfjs-converter@1.2.2":
656-
version "1.2.2"
657-
resolved "https://registry.yarnpkg.com/@tensorflow/tfjs-converter/-/tfjs-converter-1.2.2.tgz#c95e2f79b1de830b8079c7704dc8463ced2d2b79"
658-
integrity sha512-NM2NcPRHpCNeJdBxHcYpmW9ZHTQ2lJFJgmgGpQ8CxSC9CtQB05bFONs3SKcwMNDE/69QBRVom5DYqLCVUg+A+g==
655+
"@tensorflow/tfjs-converter@1.3.2":
656+
version "1.3.2"
657+
resolved "https://registry.yarnpkg.com/@tensorflow/tfjs-converter/-/tfjs-converter-1.3.2.tgz#9c75b75edbd56c736735f3ea3d3c53d1e1a2751c"
658+
integrity sha512-HTGT5sinSzayH/RXwoeebtc5pwTU9E3JzlcLfBa0HZVkkeKadhtzqh3QH+BFuuNsvX691AiIHv4P7SylUu1XQg==
659659

660-
"@tensorflow/tfjs-core@1.2.2":
661-
version "1.2.2"
662-
resolved "https://registry.yarnpkg.com/@tensorflow/tfjs-core/-/tfjs-core-1.2.2.tgz#2efa89e323612a26aeccee9b3ae9f5ac5a635bbe"
663-
integrity sha512-2hCHMKjh3UNpLEjbAEaurrTGJyj/KpLtMSAraWgHA1vGY0kmk50BBSbgCDmXWUVm7lyh/SkCq4/GrGDZktEs3g==
660+
"@tensorflow/tfjs-core@1.3.2":
661+
version "1.3.2"
662+
resolved "https://registry.yarnpkg.com/@tensorflow/tfjs-core/-/tfjs-core-1.3.2.tgz#7dcf66510e404a6564f7f47cd854eff936b58c5a"
663+
integrity sha512-42WlbkbD10F11qN7k+GMjdPB62DGGnz7czkvk+lT501qWUrQI8swTjZfgMijcTWyQSeJuLa97EOL9n/Gn3Yfpw==
664664
dependencies:
665665
"@types/offscreencanvas" "~2019.3.0"
666666
"@types/seedrandom" "2.4.27"
667667
"@types/webgl-ext" "0.0.30"
668668
"@types/webgl2" "0.0.4"
669669
node-fetch "~2.1.2"
670670
seedrandom "2.4.3"
671-
optionalDependencies:
672-
rollup-plugin-visualizer "~1.1.1"
673671

674-
"@tensorflow/tfjs-data@1.2.2":
675-
version "1.2.2"
676-
resolved "https://registry.yarnpkg.com/@tensorflow/tfjs-data/-/tfjs-data-1.2.2.tgz#bd802b4096df04277d302d66598aef47fbffef85"
677-
integrity sha512-oHGBoGdnCl2RyouLKplQqo+iil0iJgPbi/aoHizhpO77UBuJXlKMblH8w5GbxVAw3hKxWlqzYpxPo6rVRgehNA==
672+
"@tensorflow/tfjs-data@1.3.2":
673+
version "1.3.2"
674+
resolved "https://registry.yarnpkg.com/@tensorflow/tfjs-data/-/tfjs-data-1.3.2.tgz#311c510213b1ba26ebc1d989e21ae0af3feac8be"
675+
integrity sha512-1Ed6rb1OAcQc70jQqAN6UYch56p7FHafQ4cDzEDoS1tgfn0Of6Gb3KETa0RJMIpWvlrr/Oa5qwqDk8Tlusn7+Q==
678676
dependencies:
679677
"@types/node-fetch" "^2.1.2"
680678
node-fetch "~2.1.2"
681679

682-
"@tensorflow/tfjs-layers@1.2.2":
683-
version "1.2.2"
684-
resolved "https://registry.yarnpkg.com/@tensorflow/tfjs-layers/-/tfjs-layers-1.2.2.tgz#3365dbbca7cfa4fcc6cacc9fffc90d664606bd4e"
685-
integrity sha512-yzWZaZrCVpEyTkSrzMe4OOP4aGUfaaROE/zR9fPsPGGF8wLlbLNZUJjeYUmjy3G3pXGaM0mQUbLR5Vd707CVtQ==
680+
"@tensorflow/tfjs-layers@1.3.2":
681+
version "1.3.2"
682+
resolved "https://registry.yarnpkg.com/@tensorflow/tfjs-layers/-/tfjs-layers-1.3.2.tgz#1691513bce5d49f95c0aad9ee3a22eda5742828e"
683+
integrity sha512-Y+zsoXlXWYopv+u3woMgE18TsEtzcRUVdgMQxuQm1FGRFAltyKxHzVO44b/nVdC59kRHIv0wrLU4G2hSQlprqA==
686684

687-
"@tensorflow/tfjs-node-gpu@^1.2.3":
688-
version "1.2.3"
689-
resolved "https://registry.yarnpkg.com/@tensorflow/tfjs-node-gpu/-/tfjs-node-gpu-1.2.3.tgz#3786d814bc5ca4c10e88a4a490feea65a39bd8cf"
690-
integrity sha512-y8A1dF4WZZ+IvCCv/hrEUVV9O1ua0f5rZVzaMnJx+xv8o51DwTGk7h6tsnE/F2N6pf9mKLsY8roUBviIasVEmQ==
685+
"@tensorflow/tfjs-node-gpu@^1.3.2":
686+
version "1.3.2"
687+
resolved "https://registry.yarnpkg.com/@tensorflow/tfjs-node-gpu/-/tfjs-node-gpu-1.3.2.tgz#38453adb5eb677f7316c00546ba4d1c9d8c69c5b"
688+
integrity sha512-1xhwxzLo3Ae7FtWlt4l1Stb+oVWp6Jm0dISbz7vd0ZKnahOxm0oGW88ZQ9UG42jlwDB4qPlklsQMOR/+oDwH4w==
691689
dependencies:
692-
"@tensorflow/tfjs" "~1.2.2"
690+
"@tensorflow/tfjs" "1.3.2"
693691
adm-zip "^0.4.11"
694-
bindings "~1.3.0"
692+
google-protobuf "^3.9.2"
695693
https-proxy-agent "^2.2.1"
694+
node-pre-gyp "0.13.0"
696695
progress "^2.0.0"
697696
rimraf "^2.6.2"
698697
tar "^4.4.6"
699698

700-
"@tensorflow/tfjs-node@^1.2.3":
701-
version "1.2.3"
702-
resolved "https://registry.yarnpkg.com/@tensorflow/tfjs-node/-/tfjs-node-1.2.3.tgz#b2a6c3051da080a853be34b4bdc6649479139852"
703-
integrity sha512-6/V3JfoxnvUJhZle8+7V0ln7KjUIJOlDCk43EBQg+XoGudvp3L1x0RXcfCQ1nXFIlZVYixNJYd3XTIOHZBECSA==
699+
"@tensorflow/tfjs-node@^1.3.2":
700+
version "1.3.2"
701+
resolved "https://registry.yarnpkg.com/@tensorflow/tfjs-node/-/tfjs-node-1.3.2.tgz#98e71faeefac1284610e8d3bcd4c1ef36ea28255"
702+
integrity sha512-inWEd+WmVUU/UdLjYSNugTAa2MBuKUd6QtcGDk8u1+yV54/ulxYQlsITB4g/6KPnABlnpM49J342pt+/j2Z8uA==
704703
dependencies:
705-
"@tensorflow/tfjs" "~1.2.2"
704+
"@tensorflow/tfjs" "1.3.2"
706705
adm-zip "^0.4.11"
707-
bindings "~1.3.0"
706+
google-protobuf "^3.9.2"
708707
https-proxy-agent "^2.2.1"
708+
node-pre-gyp "0.13.0"
709709
progress "^2.0.0"
710710
rimraf "^2.6.2"
711711
tar "^4.4.6"
712712

713-
"@tensorflow/tfjs@1.2.2", "@tensorflow/tfjs@~1.2.2":
714-
version "1.2.2"
715-
resolved "https://registry.yarnpkg.com/@tensorflow/tfjs/-/tfjs-1.2.2.tgz#e0cc7f1c4139e7c38f3ea478999f0972d354c948"
716-
integrity sha512-HfhSzL2eTWhlT0r/A5wmo+u3bHe+an16p5wsnFH3ujn21fQ8QtGpSfDHQZjWx1kVFaQnV6KBG+17MOrRHoHlLA==
713+
"@tensorflow/tfjs@1.3.2":
714+
version "1.3.2"
715+
resolved "https://registry.yarnpkg.com/@tensorflow/tfjs/-/tfjs-1.3.2.tgz#1a7868a68a52a335e84adea9d544cc150bf86116"
716+
integrity sha512-2QrKHtM6DraAgWL9DJwluz+/CJ5EbPN9tdNAFGq83fSUm3cNEK37UPsWdQ1QXXBqMczg1uyLEmPgFq6Lsy2/Vg==
717717
dependencies:
718-
"@tensorflow/tfjs-converter" "1.2.2"
719-
"@tensorflow/tfjs-core" "1.2.2"
720-
"@tensorflow/tfjs-data" "1.2.2"
721-
"@tensorflow/tfjs-layers" "1.2.2"
718+
"@tensorflow/tfjs-converter" "1.3.2"
719+
"@tensorflow/tfjs-core" "1.3.2"
720+
"@tensorflow/tfjs-data" "1.3.2"
721+
"@tensorflow/tfjs-layers" "1.3.2"
722722

723723
"@types/node-fetch@^2.1.2":
724724
version "2.1.4"
@@ -1513,11 +1513,6 @@ bindings@~1.2.1:
15131513
resolved "https://registry.yarnpkg.com/bindings/-/bindings-1.2.1.tgz#14ad6113812d2d37d72e67b4cacb4bb726505f11"
15141514
integrity sha1-FK1hE4EtLTfXLme0ystLtyZQXxE=
15151515

1516-
bindings@~1.3.0:
1517-
version "1.3.0"
1518-
resolved "https://registry.yarnpkg.com/bindings/-/bindings-1.3.0.tgz#b346f6ecf6a95f5a815c5839fc7cdb22502f1ed7"
1519-
integrity sha512-DpLh5EzMR2kzvX1KIlVC0VkC3iZtHKTgdtZ0a3pglBZdaQFjt5S9g9xd1lE+YvXyfd6mtCeRnrUfOLYiTMlNSw==
1520-
15211516
bn.js@^4.0.0, bn.js@^4.1.0, bn.js@^4.1.1, bn.js@^4.4.0:
15221517
version "4.11.8"
15231518
resolved "https://registry.yarnpkg.com/bn.js/-/bn.js-4.11.8.tgz#2cde09eb5ee341f484746bb0309b3253b1b1442f"
@@ -2964,6 +2959,11 @@ globby@^5.0.0:
29642959
pify "^2.0.0"
29652960
pinkie-promise "^2.0.0"
29662961

2962+
google-protobuf@^3.9.2:
2963+
version "3.11.0"
2964+
resolved "https://registry.yarnpkg.com/google-protobuf/-/google-protobuf-3.11.0.tgz#9449d6029569aa4603fdaa154ec6a7681aea0787"
2965+
integrity sha512-Pni0ZGKZc/FpaRiz2n11+FfjzeuoUpnMdtuqoBUandjsEPsLDLh5C0yacl/bgNAFaZkyUgIGcGowW34u6mVWPw==
2966+
29672967
graceful-fs@^4.1.11, graceful-fs@^4.1.2, graceful-fs@^4.1.6, graceful-fs@^4.1.9:
29682968
version "4.1.15"
29692969
resolved "https://registry.yarnpkg.com/graceful-fs/-/graceful-fs-4.1.15.tgz#ffb703e1066e8a0eeaa4c8b80ba9253eeefbfb00"
@@ -3941,6 +3941,22 @@ node-libs-browser@^2.0.0:
39413941
util "^0.10.3"
39423942
vm-browserify "0.0.4"
39433943

3944+
3945+
version "0.13.0"
3946+
resolved "https://registry.yarnpkg.com/node-pre-gyp/-/node-pre-gyp-0.13.0.tgz#df9ab7b68dd6498137717838e4f92a33fc9daa42"
3947+
integrity sha512-Md1D3xnEne8b/HGVQkZZwV27WUi1ZRuZBij24TNaZwUPU3ZAFtvT6xxJGaUVillfmMKnn5oD1HoGsp2Ftik7SQ==
3948+
dependencies:
3949+
detect-libc "^1.0.2"
3950+
mkdirp "^0.5.1"
3951+
needle "^2.2.1"
3952+
nopt "^4.0.1"
3953+
npm-packlist "^1.1.6"
3954+
npmlog "^4.0.2"
3955+
rc "^1.2.7"
3956+
rimraf "^2.6.1"
3957+
semver "^5.3.0"
3958+
tar "^4"
3959+
39443960
node-pre-gyp@^0.10.0:
39453961
version "0.10.3"
39463962
resolved "https://registry.yarnpkg.com/node-pre-gyp/-/node-pre-gyp-0.10.3.tgz#3070040716afdc778747b61b6887bf78880b80fc"
@@ -4140,13 +4156,6 @@ opn@^5.1.0:
41404156
dependencies:
41414157
is-wsl "^1.1.0"
41424158

4143-
opn@^5.4.0:
4144-
version "5.5.0"
4145-
resolved "https://registry.yarnpkg.com/opn/-/opn-5.5.0.tgz#fc7164fab56d235904c51c3b27da6758ca3b9bfc"
4146-
integrity sha512-PqHpggC9bLV0VeWcdKhkpxY+3JTzetLSqTCWL/z/tFIbI6G8JCjondXklT1JinczLz2Xib62sSp0T/gKT4KksA==
4147-
dependencies:
4148-
is-wsl "^1.1.0"
4149-
41504159
optionator@^0.8.1:
41514160
version "0.8.2"
41524161
resolved "https://registry.yarnpkg.com/optionator/-/optionator-0.8.2.tgz#364c5e409d3f4d6301d6c0b4c05bba50180aeb64"
@@ -5341,16 +5350,6 @@ ripemd160@^2.0.0, ripemd160@^2.0.1:
53415350
hash-base "^3.0.0"
53425351
inherits "^2.0.1"
53435352

5344-
rollup-plugin-visualizer@~1.1.1:
5345-
version "1.1.1"
5346-
resolved "https://registry.yarnpkg.com/rollup-plugin-visualizer/-/rollup-plugin-visualizer-1.1.1.tgz#454ae0aed23845407ebfb81cc52114af308d6d90"
5347-
integrity sha512-7xkSKp+dyJmSC7jg2LXqViaHuOnF1VvIFCnsZEKjrgT5ZVyiLLSbeszxFcQSfNJILphqgAEmWAUz0Z4xYScrRw==
5348-
dependencies:
5349-
mkdirp "^0.5.1"
5350-
opn "^5.4.0"
5351-
source-map "^0.7.3"
5352-
typeface-oswald "0.0.54"
5353-
53545353
safe-buffer@^5.0.1, safe-buffer@^5.1.0, safe-buffer@^5.1.1, safe-buffer@^5.1.2, safe-buffer@~5.1.0, safe-buffer@~5.1.1:
53555354
version "5.1.2"
53565355
resolved "https://registry.yarnpkg.com/safe-buffer/-/safe-buffer-5.1.2.tgz#991ec69d296e0313747d59bdfd2b745c35f8828d"
@@ -5587,11 +5586,6 @@ source-map@^0.5.0, source-map@^0.5.3, source-map@^0.5.6, source-map@^0.5.7:
55875586
resolved "https://registry.yarnpkg.com/source-map/-/source-map-0.5.7.tgz#8a039d2d1021d22d1ea14c80d8ea468ba2ef3fcc"
55885587
integrity sha1-igOdLRAh0i0eoUyA2OpGi6LvP8w=
55895588

5590-
source-map@^0.7.3:
5591-
version "0.7.3"
5592-
resolved "https://registry.yarnpkg.com/source-map/-/source-map-0.7.3.tgz#5302f8169031735226544092e64981f751750383"
5593-
integrity sha512-CkCj6giN3S+n9qrYiBTX5gystlENnRW5jZeNLHpe6aue+SrHcG5VYwujhW9s4dY31mEGsxBDrHR6oI69fTXsaQ==
5594-
55955589
spdx-correct@^3.0.0:
55965590
version "3.0.2"
55975591
resolved "https://registry.yarnpkg.com/spdx-correct/-/spdx-correct-3.0.2.tgz#19bb409e91b47b1ad54159243f7312a858db3c2e"
@@ -5938,11 +5932,6 @@ typedarray@^0.0.6:
59385932
resolved "https://registry.yarnpkg.com/typedarray/-/typedarray-0.0.6.tgz#867ac74e3864187b1d3d47d996a78ec5c8830777"
59395933
integrity sha1-hnrHTjhkGHsdPUfZlqeOxciDB3c=
59405934

5941-
5942-
version "0.0.54"
5943-
resolved "https://registry.yarnpkg.com/typeface-oswald/-/typeface-oswald-0.0.54.tgz#1e253011622cdd50f580c04e7d625e7f449763d7"
5944-
integrity sha512-U1WMNp4qfy4/3khIfHMVAIKnNu941MXUfs3+H9R8PFgnoz42Hh9pboSFztWr86zut0eXC8byalmVhfkiKON/8Q==
5945-
59465935
unicode-canonical-property-names-ecmascript@^1.0.4:
59475936
version "1.0.4"
59485937
resolved "https://registry.yarnpkg.com/unicode-canonical-property-names-ecmascript/-/unicode-canonical-property-names-ecmascript-1.0.4.tgz#2619800c4c825800efdd8343af7dd9933cbe2818"

0 commit comments

Comments
 (0)