Shauray Singh

Training ResNet_50v1.5 on ImageNet

12

* Using SGD to train ResNet * Implement cross-entropy Loss * Adding Transforms Should reach the desired accuracy within 50 epochs in theory ( Implementation details - https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Classification/ConvNets/resnet50v1.5 ) *...

exclude float64 tests

1

This excludes all the tests that require float64. This solves https://github.com/geohot/tinygrad/issues/891#issue-1736175051

fix type annotations for arguments in training_args

3

# What does this PR do? Fixes #24538 which is basically fixing type annotations for `fsdp`, `fsdp_config`, and `sharded_ddp` in training_args.py ## Who can review? maybe @sgugger

[WIP] [Core] Add support for ControlNet LoRA

8

# What does this PR do? Adds support for ControlNet LoRA. continuing on @sayakpaul design #4899 Fixes #5800 ## Before submitting - [ ] This PR fixes a typo or...