search

jeffheaton / t81_558_deep_learning

T81-558: Keras - Applications of Deep Neural Networks @Washington University in St. Louis
https://sites.wustl.edu/jeffheaton/t81-558/
Other
5.71k stars 3.04k forks source link

t81_558_class_07_2_train_gan.ipynb - Can't install tensorflow 1.8 anymore #174

Open LeandreSassi opened 8 months ago

LeandreSassi commented 8 months ago

Dear Jeff,

When I try to run the Notebook in order to train my StyleGAN2 - ADA - Pytorch I get stuck after the first experiment. It produces the first generated .png file, however it doesnt make any .pkl file. I think it might be because I run a too young Torch version as Colab doesnt allow me to downgrade Pytorch to 1.8 in the second cell of this notebook

Here is the log:

ERROR: Could not find a version that satisfies the requirement torch==1.8.1 (from versions: 1.11.0, 1.12.0, 1.12.1, 1.13.0, 1.13.1, 2.0.0, 2.0.1, 2.1.0, 2.1.1, 2.1.2) ERROR: No matching distribution found for torch==1.8.1

Is there any way the problems are related? If so, is there any workaround? Or any newer Notebook on training a StyleGAN2 or 3 with our own dataset?

thank you in advance!

Leandre Sassi

LeandreSassi commented 8 months ago

Update:

You can find the solution here https://github.com/dvschultz/stylegan2-ada-pytorch/issues/45#issuecomment-1671468176