Update README.md
Browse files
README.md
CHANGED
|
@@ -16,7 +16,7 @@ pipeline_tag: text-to-image
|
|
| 16 |
|
| 17 |
```
|
| 18 |
This is the config for the PT, you want the student unless you want to train it
|
| 19 |
-
I KNOW I KNOW, I'll get it worked out. For now this is epoch 9.
|
| 20 |
|
| 21 |
"cfg": asdict(self.cfg),
|
| 22 |
"student": self.student.state_dict(),
|
|
|
|
| 16 |
|
| 17 |
```
|
| 18 |
This is the config for the PT, you want the student unless you want to train it
|
| 19 |
+
I KNOW I KNOW, I'll get it worked out. For now this is every epoch 9+ if you see a PT for this particular model.
|
| 20 |
|
| 21 |
"cfg": asdict(self.cfg),
|
| 22 |
"student": self.student.state_dict(),
|