ProGamerGov / neural-style-pt

PyTorch implementation of neural style transfer algorithm
MIT License
838 stars 175 forks source link

add 'transfer' function that can be called from another script #63

Open momo-the-monster opened 4 years ago

momo-the-monster commented 4 years ago

also adds TransferParams class that incorporates all known command-line params.

arthur-fontaine commented 3 years ago

@ProGamerGov When are you going to merge? I'd like to use it a function...

lucasfijen commented 3 years ago

Please note that some of the functions, such as setup_multi_device(), setup_gpu() and setup_optimizer still rely on the global params settings in this PR, as they are defined outside of the transfer().

I think that could be best fixed by placing the argparser into the main() function, and adding the params variables in the functions that are outside the transfer(params) function