Updated model and checkpoints to run in tf v2.#127
Updated model and checkpoints to run in tf v2.#127brunomcebola wants to merge 1 commit intogoogle-deepmind:masterfrom
Conversation
|
Thanks! Do we need to have the data folder copied into both tf1 and tf2 folders or would it be better to keep it shared ? Did you test both versions ? |
|
Regarding the data folder, the actual data (.txt, .npy and .gif) could be shared accross both versions. The checkpoints, however, are version specific and should be left inside the respective folder for organization sake. Nonetheless, since the access to this data is performed by setting the constants at the beginning of the scripts (making it possible for this folder to anywhere on the user machine), I choose to keep it separate just be more user friendly for anyone wanting to use the code. Regarding the test of both versions, yes, I tested them and both worked in my machine with GPU support. An important note however are the subversions of tf, which need to be chosen according to the cuda versions, but this does not afect the final code of each version at all. Hope this has been clear! |
Changes:
Folder Structure:
tf_v1.tf_v2to accommodate the updated version.Migration:
tf_v2folder.Environment Setup:
requirements.txtand.python-versionfiles to both thetf_v1andtf_v2folders to ensure proper environment setup and compatibility with respective TensorFlow versions.