-
Notifications
You must be signed in to change notification settings - Fork 8
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Add boilerplate code for Dataflux-Pytorch Lightning demo #57
Conversation
Add a new flag for specifying number of dataloader threads (GoogleCloudPlatform#37)
Add boilerplate code for Dataflux-Pytorch lightning demo
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Moved the three new files from demo/lightning to demo/lightning/image-segmentation. Will probably have to move demo/image-segmentation to demo/pytorch/image-segmentation in a later PR.
I agree that the arguments to Unet3D and Unet3DDataModule constructors should be configurable. I will make that change after I have this workload up and running with hardcoded parameters.
No description provided.