search

FrozenBurning / Text2Light

[SIGGRAPH Asia 2022] Text2Light: Zero-Shot Text-Driven HDR Panorama Generation
https://frozenburning.github.io/projects/text2light/
Other
577 stars 46 forks source link

Is it possible to do all the training on colab? #15

Closed taginterco closed 1 year ago

taginterco commented 1 year ago

I want to train with my own dataset and I preprocessed and everything, but its been tricky doing it all on colab because the trainingstage1.py is using different versions of pytorch lightning, does it have to be only in a conda environment?

FrozenBurning commented 1 year ago

I haven't tried to train the model using colab. And I'm afraid that the training would be slow given the limited capacity of GPU. The environment is not necessarily in conda at all. You can install all dependencies via pip according to environment.yml.

taginterco commented 1 year ago

Thank you!

taginterco commented 1 year ago

I tried installing the dependencies manually using pip but the versions sort of conflict with each other, such as there is a module used in pytorch version 1.7 and another one used in pytorch 1.9 but I can only install one of them.